Searched defs:seq_dim (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | reverse_sequence_op.cc | 44 void CheckErrors(OpKernelContext* context, int batch_dim, int seq_dim) { in CheckErrors()
78 void CheckErrorsGPU(OpKernelContext* context, int batch_dim, int seq_dim) { in CheckErrorsGPU()
99 int seq_dim) { in CheckErrors()
105 int seq_dim) { in CheckErrors()
|
| D | reverse_sequence_op.h | 33 int32 seq_dim, typename TTypes<Tlen>::ConstVec seq_lengths) in ReverseGenerator()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | reverse_sequence.cc | 72 int seq_dim = params->seq_dim; in ReverseSequenceImpl() local
|
| D | reverse_sequence_test.cc | 30 int seq_dim, int batch_dim) { in ReverseSequenceOpModel()
|
| /external/tensorflow/tensorflow/contrib/recurrent/python/ops/ |
| D | functional_rnn.py | 403 def _reverse(input_, seq_lengths, seq_dim, batch_dim): argument
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad.cc | 248 int seq_dim; in ReverseSequenceGrad() local
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | builtin_op_data.h | 372 int seq_dim; member
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops_test.cc | 1111 auto rebuild_node_def = [&op](const int32 seq_dim, const int32 batch_dim) { in TEST()
|
| D | array_ops.cc | 1480 int64 seq_dim; in __anon7c94107b2402() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_ops.py | 3247 seq_dim=None, argument
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 2742 int seq_dim = shuffled_dim_ids[1]; in TEST_F() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 4685 void ReverseSequence(const TS* seq_lengths, const int seq_dim, in ReverseSequence()
|