Searched defs:_batch_sizes (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | PackedSequence.cpp | 115 …ard_symint(const Tensor& grad, c10::SymIntArrayRef input_size, const Tensor& _batch_sizes, bool ba… in _pack_padded_sequence_backward_symint()
142 std::tuple<Tensor, Tensor> _pad_packed_sequence(const Tensor& data, const Tensor& _batch_sizes, boo… in _pad_packed_sequence()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/miopen/ |
| H A D | RNN_miopen.cpp | 823 const Tensor& input, const Tensor& _batch_sizes, const hidden_type& hidden, in _miopen_impl()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/ |
| H A D | RNN.cpp | 2500 const Tensor& _batch_sizes, in _cudnn_impl()
|