Searched defs:seq_lens (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | padded_batch_test.py | 356 def build_dataset(seq_lens): argument
369 def build_dataset(seq_lens): argument
394 def build_dataset(seq_lens): argument
407 def build_dataset(seq_lens): argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| D | ctc_decoder_ops_test.py | 46 seq_lens, argument
|
| D | ctc_loss_op_test.py | 85 seq_lens, argument
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | reverse_sequence_op.cc | 63 const auto seq_lens = context->Input(1); in Compile() local
|
| /external/pytorch/test/onnx/ |
| D | test_pytorch_onnx_no_runtime.py | 755 def forward(self, x, seq_lens): argument
820 def forward(self, x, seq_lens): argument
871 def forward(self, x, seq_lens): argument
|
| /external/pytorch/test/inductor/ |
| D | test_cpu_select_algorithm.py | 779 self, batch_size, in_features, seq_lens, out_features, bias, dtype argument
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 3938 std::vector<int32> seq_lens(batch_size); in TEST_F() local
|
| /external/pytorch/test/ |
| D | test_jit.py | 9666 def pack_padded_pad_packed_script(x, seq_lens): argument
|