Home
last modified time | relevance | path

Searched refs:pad_packed_sequence (Results 1 – 16 of 16) sorted by relevance

/external/pytorch/test/onnx/model_defs/
Drnn_model_with_packed_sequence.py16 ret, _ = rnn_utils.pad_packed_sequence(ret, self.batch_first)
30 ret, _ = rnn_utils.pad_packed_sequence(ret, self.batch_first)
44 ret, _ = rnn_utils.pad_packed_sequence(ret, self.batch_first)
/external/pytorch/test/nn/
Dtest_packed_sequence.py62 unpacked, lengths_out = rnn_utils.pad_packed_sequence(masked)
98 rnn_utils.pad_packed_sequence(
110 no_extra_pad, _ = rnn_utils.pad_packed_sequence(
115 unpacked, lengths_out = rnn_utils.pad_packed_sequence(
296 unpacked = rnn_utils.pad_packed_sequence(packed, batch_first)
467 unpacked, unpacked_len = rnn_utils.pad_packed_sequence(
/external/pytorch/docs/source/notes/
Dfaq.rst148 :func:`torch.nn.utils.rnn.pad_packed_sequence` by default only pads up to the
152 :func:`~torch.nn.utils.rnn.pad_packed_sequence` to make sure that the
156 from torch.nn.utils.rnn import pack_padded_sequence, pad_packed_sequence
170 output, _ = pad_packed_sequence(packed_output, batch_first=True,
/external/pytorch/torch/nn/utils/
Drnn.py342 def pad_packed_sequence( function
597 padded_sequences, lengths = pad_packed_sequence(packed_sequences, batch_first=True)
/external/pytorch/test/distributed/fsdp/
Dtest_utils.py123 x, _ = nn.utils.rnn.pad_packed_sequence(x)
/external/pytorch/test/cpp/api/
Dnn_utils.cpp477 rnn_utils::pad_packed_sequence( in TEST_F()
492 rnn_utils::pad_packed_sequence(packed, /*batch_first=*/batch_first)); in TEST_F()
495 auto [unpacked, lengths_out] = rnn_utils::pad_packed_sequence( in TEST_F()
558 rnn_utils::pad_packed_sequence(packed, batch_first); in TEST_F()
753 rnn_utils::pad_packed_sequence(packed, /*batch_first=*/batch_first); in TEST_F()
Drnn.cpp811 auto error = torch::nn::utils::rnn::pad_packed_sequence(packed); in TEST_F()
/external/pytorch/test/onnx/
Dtest_pytorch_onnx_no_runtime.py750 from torch.nn.utils.rnn import pack_padded_sequence, pad_packed_sequence
757 x, _ = pad_packed_sequence(x)
799 from torch.nn.utils.rnn import pack_padded_sequence, pad_packed_sequence
823 x, _ = pad_packed_sequence(x)
857 from torch.nn.utils.rnn import pack_padded_sequence, pad_packed_sequence
880 x, _ = pad_packed_sequence(x)
/external/pytorch/torch/csrc/api/include/torch/nn/utils/
Drnn.h246 inline std::tuple<Tensor, Tensor> pad_packed_sequence(
/external/pytorch/test/cpp_api_parity/
Dparity-tracker.md149 torch::nn::utils::rnn::pad_packed_sequence|Yes|No
/external/pytorch/benchmarks/fastrnns/
Dfactory.py250 padded, lengths = rnn_utils.pad_packed_sequence(out)
/external/pytorch/docs/source/
Dnn.rst501 nn.utils.rnn.pad_packed_sequence
/external/pytorch/benchmarks/functional_autograd_benchmark/
Dtorchaudio_models.py212 x, _ = nn.utils.rnn.pad_packed_sequence(x)
/external/pytorch/test/
Dtest_expanded_weights.py760 padded_input, seq_sizes = torch.nn.utils.rnn.pad_packed_sequence(
Dtest_jit.py2503 padded_outs, _ = torch.nn.utils.rnn.pad_packed_sequence(out)
2526 padded_outs, _ = torch.nn.utils.rnn.pad_packed_sequence(out)
9664 from torch.nn.utils.rnn import pack_padded_sequence, pad_packed_sequence
9668 x, lengths = pad_packed_sequence(x)
9695 output, lengths = pad_packed_sequence(
9707 from torch.nn.utils.rnn import pad_sequence, pack_sequence, pad_packed_sequence
9715 return pad_packed_sequence(pack_sequence(tensor_list, enforce_sorted))[0]
Dtest_nn.py10831 unpacked, _ = rnn_utils.pad_packed_sequence(packed)
11329 unpacked, unpacked_len = rnn_utils.pad_packed_sequence(packed_out)