Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DFractionalMaxPool2d.cpp60 int64_t inputH = input.size(heightDim); in TORCH_META_FUNC() local
106 auto inputH = input.size(heightDim); 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()
316 int64_t inputH = input.size(heightDim); in TORCH_IMPL_FUNC() local
369 auto inputH = input.size(heightDim); in TORCH_IMPL_FUNC() local
H A DFractionalMaxPool3d.cpp67 int64_t inputH = input_.size(heightDim); 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()
356 int64_t inputH = input.size(heightDim); in fractional_max_pool3d_backward_out_cpu_template() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DFractionalMaxPool2d.cu122 int inputH = index / gradInput.size(3); in fractional_max_pool2d_backward_out_cuda_frame() local
225 int inputH = input.size(dimh); in TORCH_IMPL_FUNC() local
H A DFractionalMaxPool3d.cu142 int64_t inputH = (index / gradInput.size(4)) % in fractional_max_pool3d_backward_out_frame() local
178 int64_t inputH = input.size(dimh); in fractional_max_pool3d_backward_out_cuda_template() local
H A DReplicationPadding.cu628 const auto inputH = input.size(dimh); in TORCH_IMPL_FUNC() local