| /aosp_15_r20/external/pytorch/test/nn/ |
| H A D | test_packed_sequence.py | 55 for enforce_sorted in [True, False]:
58 padded, lengths, enforce_sorted=enforce_sorted
71 lambda: rnn_utils.pack_padded_sequence(b_a, [22, 25], enforce_sorted=True),
135 for enforce_sorted in (True, False):
138 padded, lengths, enforce_sorted=enforce_sorted
293 def _compatibility_test(sequences, lengths, batch_first, enforce_sorted=False): argument
295 packed = rnn_utils.pack_sequence(sequences, enforce_sorted)
299 padded, lengths, batch_first, enforce_sorted
307 packed = rnn_utils.pack_sequence([a, b, c], enforce_sorted=False)
314 packed_unsorted = rnn_utils.pack_sequence([b, c, a], enforce_sorted=False)
[all …]
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/utils/ |
| H A D | rnn.h | 202 bool enforce_sorted = true) {
205 if (enforce_sorted) {
339 bool enforce_sorted = true) {
348 /*enforce_sorted=*/enforce_sorted);
|
| /aosp_15_r20/external/pytorch/torch/nn/utils/ |
| H A D | rnn.py | 284 enforce_sorted: bool = True,
330 if enforce_sorted:
534 enforce_sorted: bool = True,
567 pad_sequence(sequences), lengths, enforce_sorted=enforce_sorted
|
| /aosp_15_r20/external/pytorch/test/cpp/api/ |
| H A D | nn_utils.cpp | 522 for (bool enforce_sorted : std::vector<bool>{true, false}) { in TEST_F() local
528 /*enforce_sorted=*/enforce_sorted) in TEST_F()
553 bool enforce_sorted = false) { in TEST_F() argument
556 rnn_utils::pack_sequence(sequences, enforce_sorted); in TEST_F()
561 padded, lengths, batch_first, enforce_sorted); in TEST_F()
638 for (bool enforce_sorted : std::vector<bool>{true, false}) { in TEST_F() local
640 sequences, torch::tensor(lengths_vec), batch_first, enforce_sorted); in TEST_F()
|
| /aosp_15_r20/external/pytorch/benchmarks/fastrnns/ |
| H A D | factory.py | 248 packed = rnn_utils.pack_sequence(sequences, enforce_sorted=False)
|
| /aosp_15_r20/external/pytorch/benchmarks/functional_autograd_benchmark/ |
| H A D | torchaudio_models.py | 210 x = nn.utils.rnn.pack_padded_sequence(x, output_lengths, enforce_sorted=False)
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_dataloader.py | 3319 return torch.nn.utils.rnn.pack_padded_sequence(data, lengths, enforce_sorted=False)
3327 data, lengths, batch_first=True, enforce_sorted=False
|
| H A D | test_nn.py | 10824 for enforce_sorted in [True, False]:
10827 padded, lengths, enforce_sorted=enforce_sorted)
11291 def check_lengths(lengths, enforce_sorted, use_default_hiddens, proj_size): argument
11327 packed = rnn_utils.pack_padded_sequence(x, lengths, enforce_sorted=enforce_sorted)
11359 for enforce_sorted, seq_lens, in tests:
11362 check_lengths(seq_lens, enforce_sorted, use_default_hiddens, proj_size)
|
| H A D | test_jit.py | 9693 input=input, lengths=torch.tensor([1, 2]), enforce_sorted=False
9713 def pack_sequence_func(tensor_list, enforce_sorted=True): argument
9715 return pad_packed_sequence(pack_sequence(tensor_list, enforce_sorted))[0]
|
| H A D | test_cuda.py | 5225 enforce_sorted=False,
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_cpu_repro.py | 585 sent, sent_lens, batch_first=True, enforce_sorted=False
|