| /external/tensorflow/tensorflow/core/kernels/ |
| D | reverse_sequence_op.cc | 46 const Tensor& seq_lengths = context->input(1); in CheckErrors() local
81 const Tensor& seq_lengths = context->input(1); in CheckErrorsGPU() local
126 const Tensor& seq_lengths = context->input(1); in Compute() local
|
| D | reverse_sequence_op.h | 33 int32_t seq_dim, typename TTypes<Tlen>::ConstVec seq_lengths) in ReverseGenerator()
|
| D | cudnn_rnn_ops.cc | 700 const absl::Span<const int> seq_lengths, bool time_major) { in CreateForwardAndBackwardIODescriptors()
807 absl::Span<const int> seq_lengths; in DoForward() local
890 absl::Span<const int> seq_lengths; in DoBackward() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | reverse_sequence.cc | 41 const TfLiteTensor* seq_lengths; in Prepare() local
78 const TS* seq_lengths = GetTensorData<TS>(seq_lengths_tensor); in ReverseSequenceImpl() local
|
| D | reverse_sequence_test.cc | 32 ReverseSequenceOpModel(const TensorData& input, const TensorData& seq_lengths, in ReverseSequenceOpModel()
46 int seq_lengths() { return seq_lengths_; } in seq_lengths() function in tflite::__anona28d4b210111::ReverseSequenceOpModel
|
| /external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| D | reverse_sequence_op_test.py | 37 seq_lengths, argument
59 seq_lengths, argument
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | reverse_sequence_op_test.py | 31 seq_lengths, argument
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | rnn.py | 456 def _reverse(input_, seq_lengths, seq_axis, batch_axis): argument
|
| D | ctc_ops.py | 556 def ctc_state_log_probs(seq_lengths, max_seq_length): argument
|
| D | array_ops.py | 4980 seq_lengths, argument
5041 seq_lengths, argument
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad.cc | 268 auto seq_lengths = op.input(1); in ReverseSequenceGrad() local
|
| D | array_grad_test.cc | 219 auto seq_lengths = Const(scope_, {1, 2, 3, 4, 5}); in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | stream_executor_pimpl.cc | 409 const absl::Span<const int>& seq_lengths, bool time_major, in createRnnSequenceTensorDescriptor()
|
| D | dnn.h | 2069 const absl::Span<const int>& seq_lengths, in createRnnSequenceTensorDescriptor()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 915 void ReverseSequence(const TS* seq_lengths, const int seq_dim, in ReverseSequence()
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1672 const absl::Span<const int>& seq_lengths, bool time_major, in Create()
2450 const absl::Span<const int>& seq_lengths, bool time_major, in createRnnSequenceTensorDescriptor()
|