Home
last modified time | relevance | path

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

/external/pytorch/torch/csrc/api/src/nn/modules/
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
/external/pytorch/aten/src/ATen/native/cuda/
DReflectionPad.cu599 auto output_packed = output_inner.packed_accessor64<scalar_t, 5>(); in TORCH_IMPL_FUNC() local