| /external/tensorflow/tensorflow/lite/kernels/ |
| D | reverse_sequence.cc | 37 const TfLiteTensor* seq_lengths = GetInput(context, node, kSeqLengthsTensor); in Prepare() local
68 const TS* seq_lengths = GetTensorData<TS>(seq_lengths_tensor); in ReverseSequenceImpl() local
|
| D | reverse_sequence_test.cc | 29 ReverseSequenceOpModel(const TensorData& input, const TensorData& seq_lengths, in ReverseSequenceOpModel()
43 int seq_lengths() { return seq_lengths_; } in seq_lengths() function in tflite::__anond053f1e50111::ReverseSequenceOpModel
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | reverse_sequence_op_test.py | 38 seq_lengths, argument
57 seq_lengths, argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | reverse_sequence_op.h | 33 int32 seq_dim, typename TTypes<Tlen>::ConstVec seq_lengths) in ReverseGenerator()
|
| D | cudnn_rnn_ops.cc | 655 const absl::Span<const int>& seq_lengths, bool time_major) { in CreateForwardAndBackwardIODescriptors()
745 absl::Span<const int> seq_lengths; in DoForward() local
807 absl::Span<const int> seq_lengths; in DoBackward() local
|
| /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 | 389 def _reverse(input_, seq_lengths, seq_axis, batch_axis): argument
|
| /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/python/ops/ |
| D | ctc_ops.py | 415 def ctc_state_log_probs(seq_lengths, max_seq_length): argument
|
| D | rnn.py | 453 def _reverse(input_, seq_lengths, seq_axis, batch_axis): argument
|
| D | array_ops.py | 3243 seq_lengths, argument
3269 input, seq_lengths, seq_axis=None, batch_axis=None, name=None): argument
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad.cc | 245 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 | 414 const absl::Span<const int> &seq_lengths, bool time_major, in createRnnSequenceTensorDescriptor()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1296 const absl::Span<const int>& seq_lengths, bool time_major, in Create()
1852 const absl::Span<const int>& seq_lengths, bool time_major, in createRnnSequenceTensorDescriptor()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 4685 void ReverseSequence(const TS* seq_lengths, const int seq_dim, in ReverseSequence()
|