Searched refs:seq_dim (Results 1 - 8 of 8) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
H A D | reverse_sequence_op.cc | 44 void CheckErrors(OpKernelContext* context, int batch_dim, int seq_dim) { argument 56 OP_REQUIRES(context, batch_dim != seq_dim, 57 errors::InvalidArgument("batch_dim == seq_dim == ", seq_dim)); 58 OP_REQUIRES(context, seq_dim < input.dims(), 59 errors::InvalidArgument("seq_dim must be < input.dims()", "( ", 60 seq_dim, " vs. ", input.dims(), ")")); 72 OP_REQUIRES(context, seq_lens_vec[d] <= input.dim_size(seq_dim), 74 seq_dim, ")")); 78 void CheckErrorsGPU(OpKernelContext* context, int batch_dim, int seq_dim) { argument 98 CheckErrors(OpKernelContext* context, int batch_dim, int seq_dim) argument 104 CheckErrors(OpKernelContext* context, int batch_dim, int seq_dim) argument [all...] |
H A D | reverse_sequence_op.h | 33 int32 seq_dim, typename TTypes<Tlen>::ConstVec seq_lengths) 36 seq_dim_(seq_dim), 65 int32 batch_dim, int32 seq_dim, 69 seq_dim, seq_lengths); 32 ReverseGenerator(typename TTypes<T, Dims>::ConstTensor input, int32 batch_dim, int32 seq_dim, typename TTypes<Tlen>::ConstVec seq_lengths) argument 63 Compute( const Device& d, typename TTypes<T, Dims>::ConstTensor input, int32 batch_dim, int32 seq_dim, typename TTypes<Tlen>::ConstVec seq_lengths, typename TTypes<T, Dims>::Tensor output) argument
|
/external/tensorflow/tensorflow/python/ops/ |
H A D | rnn.py | 429 def _reverse(input_, seq_lengths, seq_dim, batch_dim): 433 seq_dim=seq_dim, batch_dim=batch_dim) 435 return array_ops.reverse(input_, axis=[seq_dim]) 440 seq_dim=time_dim, batch_dim=batch_dim) 449 seq_dim=time_dim, batch_dim=batch_dim)
|
H A D | array_ops.py | 2640 seq_dim=None, 2643 "seq_dim", seq_dim) 2649 seq_dim=seq_axis, 2659 "seq_dim", "seq_axis")
|
/external/tensorflow/tensorflow/cc/gradients/ |
H A D | array_grad.cc | 230 int seq_dim; local 231 TF_RETURN_IF_ERROR(GetNodeAttr(op.node()->attrs(), "seq_dim", &seq_dim)); 233 ReverseSequence(scope, grad_inputs[0], seq_lengths, seq_dim,
|
/external/tensorflow/tensorflow/contrib/crf/python/ops/ |
H A D | crf.py | 490 backpointers, sequence_length - 1, seq_dim=1) 508 decode_tags, sequence_length, seq_dim=1)
|
/external/tensorflow/tensorflow/core/ops/ |
H A D | array_ops_test.cc | 1060 auto rebuild_node_def = [&op](const int32 seq_dim, const int32 batch_dim) { 1064 .Attr("seq_dim", seq_dim) 1076 // Validate seq_dim and batch_dim 1080 INFER_ERROR("seq_dim must be < input rank", op, "[1,2,3];[3]");
|
H A D | array_ops.cc | 1242 .Attr("seq_dim: int") 1251 int64 seq_dim; 1252 TF_RETURN_IF_ERROR(c->GetAttr("seq_dim", &seq_dim)); 1260 // Validate batch_dim and seq_dim against input. 1266 if (seq_dim >= input_rank) { 1268 "seq_dim must be < input rank: ", seq_dim, " vs. ", input_rank);
|
Completed in 351 milliseconds