Searched defs:seq_dim (Results 1 – 11 of 11) 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()
79 void CheckErrorsGPU(OpKernelContext* context, int batch_dim, int seq_dim) { in CheckErrorsGPU()
101 int seq_dim) { in CheckErrors()
107 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 | 82 int seq_dim = params->seq_dim; in ReverseSequenceImpl() local
|
| D | reverse_sequence_test.cc | 33 int seq_dim, int batch_dim) { in ReverseSequenceOpModel()
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad.cc | 257 int seq_dim; in ReverseSequenceGrad() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops_test.cc | 1146 auto rebuild_node_def = [&op](const int32 seq_dim, const int32 batch_dim) { in TEST()
|
| D | array_ops.cc | 1619 int64 seq_dim; in __anon42d741192402() local
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | builtin_op_data.h | 460 int seq_dim; member
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_ops.py | 4708 seq_dim=None, argument
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 1720 void ReverseSequence(const TS* seq_lengths, const int seq_dim, in ReverseSequence()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 2743 int seq_dim = shuffled_dim_ids[1]; in TEST_F() local
|