Home
last modified time | relevance | path

Searched defs:inputW (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DFractionalMaxPool2d.cpp61 auto inputW = input.size(widthDim); in TORCH_META_FUNC() local
107 auto inputW = input.size(widthDim); in TORCH_META_FUNC() local
136 int inputW, int inputH, in fractional_max_pool2d_out_single_batch_frame()
197 int inputW, int inputH, in fractional_max_pool2d_out_frame()
228 int inputW, int inputH, in fractional_max_pool2d_backward_out_single_batch_frame()
255 int inputW, int inputH, in fractional_max_pool2d_backward_out_frame()
317 int64_t inputW = input.size(widthDim); in TORCH_IMPL_FUNC() local
370 auto inputW = input.size(widthDim); in TORCH_IMPL_FUNC() local
H A DFractionalMaxPool3d.cpp68 int64_t inputW = input_.size(widthDim); in TORCH_PRECOMPUTE_META_FUNC() local
108 int64_t inputT, int64_t inputH, int64_t inputW, in fractional_max_pool3d_out_single_batch_frame()
180 int64_t inputT, int64_t inputH, int64_t inputW, in fractional_max_pool3d_out_frame()
267 int64_t inputT, int64_t inputH, int64_t inputW, in fractional_max_pool3d_backward_out_single_batch_frame()
299 int64_t inputT, int64_t inputH, int64_t inputW, in fractional_max_pool3d_backward_out_frame()
357 int64_t inputW = input.size(widthDim); in fractional_max_pool3d_backward_out_cpu_template() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DFractionalMaxPool2d.cu121 int inputW = index % gradInput.size(3); in fractional_max_pool2d_backward_out_cuda_frame() local
226 int inputW = input.size(dimw); in TORCH_IMPL_FUNC() local
H A DFractionalMaxPool3d.cu141 int64_t inputW = index % gradInput.size(4); in fractional_max_pool3d_backward_out_frame() local
179 int64_t inputW = input.size(dimw); in fractional_max_pool3d_backward_out_cuda_template() local
H A DReplicationPadding.cu433 int64_t inputW = input.size(dimw); in TORCH_IMPL_FUNC() local
629 const auto inputW = input.size(dimw); in TORCH_IMPL_FUNC() local