Home
last modified time | relevance | path

Searched defs:output_packed (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/modules/
H A Drnn.cpp543 auto output_packed = in forward_with_packed_input() local
694 auto output_packed = in forward_with_packed_input() local
790 auto output_packed = in forward_with_packed_input() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DReflectionPad.cu599 auto output_packed = output_inner.packed_accessor64<scalar_t, 5>(); in TORCH_IMPL_FUNC() local