Searched full:_pad_packed_sequence (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | PackedSequence.cpp | 142 std::tuple<Tensor, Tensor> _pad_packed_sequence(const Tensor& data, const Tensor& _batch_sizes, boo… in _pad_packed_sequence() function
|
| H A D | native_functions.yaml | 7845 - func: _pad_packed_sequence(Tensor data, Tensor batch_sizes, bool batch_first, Scalar padding_valu…
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/utils/ |
| H A D | rnn.h | 264 auto [padded_output, lengths] = torch::_pad_packed_sequence(
|
| /aosp_15_r20/external/pytorch/test/mobile/model_test/ |
| H A D | model_ops.yaml | 39 aten::_pad_packed_sequence: 23
|
| H A D | coverage.yaml | 1056 aten::_pad_packed_sequence: 10
|
| /aosp_15_r20/external/pytorch/torch/nn/utils/ |
| H A D | rnn.py | 404 padded_output, lengths = _VF._pad_packed_sequence(
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | allowlist_for_publicAPI.json | 1279 "_pad_packed_sequence",
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset9.py | 4719 @_onnx_symbolic("aten::_pad_packed_sequence")
4721 def _pad_packed_sequence( function
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/ |
| H A D | trace_rules.py | 1538 "torch._pad_packed_sequence",
|