Searched full:_pack_padded_sequence (Results 1 – 13 of 13) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_subclasses/ |
| H A D | fake_impls.py | 721 @register_op_impl(torch.ops.aten._pack_padded_sequence.default)
722 def _pack_padded_sequence(fake_mode, func, inputs, lengths, batch_first): function
|
| /aosp_15_r20/external/pytorch/test/expect/ |
| H A D | HasDecompTest.test_has_decomposition.expect | 482 aten::_pack_padded_sequence
483 aten::_pack_padded_sequence.out
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | PackedSequence.cpp | 34 std::tuple<Tensor, Tensor> _pack_padded_sequence(const Tensor& _input, const Tensor& _lengths, bool… in _pack_padded_sequence() function
|
| H A D | native_functions.yaml | 7836 - func: _pack_padded_sequence(Tensor input, Tensor lengths, bool batch_first) -> (Tensor, Tensor)
7838 CompositeExplicitAutograd: _pack_padded_sequence
7839 autogen: _pack_padded_sequence.out
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/utils/ |
| H A D | rnn.h | 216 torch::_pack_padded_sequence(input, lengths, batch_first);
|
| /aosp_15_r20/external/pytorch/test/mobile/model_test/ |
| H A D | model_ops.yaml | 38 aten::_pack_padded_sequence: 23
|
| H A D | coverage.yaml | 1055 aten::_pack_padded_sequence: 10
|
| /aosp_15_r20/external/pytorch/torch/nn/utils/ |
| H A D | rnn.py | 338 data, batch_sizes = _VF._pack_padded_sequence(input, lengths, batch_first)
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | allowlist_for_publicAPI.json | 1278 "_pack_padded_sequence",
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset9.py | 4694 @_onnx_symbolic("aten::_pack_padded_sequence")
4696 def _pack_padded_sequence(g: jit_utils.GraphContext, input, lengths, batch_first): function
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/ |
| H A D | trace_rules.py | 1537 "torch._pack_padded_sequence",
|
| /aosp_15_r20/external/pytorch/tools/autograd/ |
| H A D | derivatives.yaml | 2869 - name: _pack_padded_sequence(Tensor input, Tensor lengths, bool batch_first) -> (Tensor, Tensor)
|
| /aosp_15_r20/external/pytorch/test/onnx/ |
| H A D | test_pytorch_onnx_onnxruntime.py | 13841 # Operator aten::_pack_padded_sequence is not supported by exporter yet.
|