Searched defs:_pack_padded_sequence (Results 1 – 3 of 3) sorted by relevance
| /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
|
| /aosp_15_r20/external/pytorch/torch/_subclasses/ |
| H A D | fake_impls.py | 722 def _pack_padded_sequence(fake_mode, func, inputs, lengths, batch_first): function
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset9.py | 4696 def _pack_padded_sequence(g: jit_utils.GraphContext, input, lengths, batch_first): function
|