| /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
122 const Tensor& seq_lengths = context->input(1); in Compute() local
|
| D | reverse_sequence_op.h | 33 int32 seq_dim, typename TTypes<Tlen>::ConstVec seq_lengths) in ReverseGenerator()
|
| D | cudnn_rnn_ops.cc | 697 const absl::Span<const int>& seq_lengths, bool time_major) { in CreateForwardAndBackwardIODescriptors()
804 absl::Span<const int> seq_lengths; in DoForward() local
867 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::__anon938e961c0111::ReverseSequenceOpModel
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | reverse_sequence_op_test.py | 38 seq_lengths, argument
57 seq_lengths, argument
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | reverse_sequence_op_test.py | 35 seq_lengths, argument
|
| /external/tensorflow/tensorflow/lite/experimental/examples/lstm/ |
| D | rnn.py | 392 def _reverse(input_, seq_lengths, seq_axis, batch_axis): argument
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | rnn.py | 457 def _reverse(input_, seq_lengths, seq_axis, batch_axis): argument
|
| D | ctc_ops.py | 510 def ctc_state_log_probs(seq_lengths, max_seq_length): argument
|
| D | array_ops.py | 4704 seq_lengths, argument
4765 seq_lengths, argument
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad.cc | 254 auto seq_lengths = op.input(1); in ReverseSequenceGrad() local
|
| D | array_grad_test.cc | 211 auto seq_lengths = Const(scope_, {1, 2, 3, 4, 5}); in TEST_F() local
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.cc | 399 const absl::Span<const int> &seq_lengths, bool time_major, in createRnnSequenceTensorDescriptor()
|
| D | dnn.h | 2141 const absl::Span<const int>& seq_lengths, in createRnnSequenceTensorDescriptor()
|
| /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/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1495 const absl::Span<const int>& seq_lengths, bool time_major, in Create()
2107 const absl::Span<const int>& seq_lengths, bool time_major, in createRnnSequenceTensorDescriptor()
|