Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DFractionalMaxPool2d.cpp88 int64_t numBatch = 1; in TORCH_META_FUNC() local
196 int numBatch, int numPlanes, in fractional_max_pool2d_out_frame()
254 int numBatch, int numPlanes, in fractional_max_pool2d_backward_out_frame()
292 int64_t numBatch = 1; in TORCH_IMPL_FUNC() local
351 int64_t numBatch = 1; in TORCH_IMPL_FUNC() local
H A DFractionalMaxPool3d.cpp40 int64_t numBatch = 1; in TORCH_PRECOMPUTE_META_FUNC() local
179 int64_t numBatch, int64_t numPlanes, in fractional_max_pool3d_out_frame()
298 int64_t numBatch, int64_t numPlanes, in fractional_max_pool3d_backward_out_frame()
338 int64_t numBatch = 1; in fractional_max_pool3d_backward_out_cpu_template() local
H A DMaxUnpooling.cpp56 int64_t numBatch = self.size(0); in max_unpooling2d_forward_out_cpu() local
/aosp_15_r20/external/intel-media-driver/media_softlet/linux/common/os/xe/
H A Dmos_gpucontext_specific_next_xe.cpp520 int numBatch = 0; in ParallelSubmitCommands() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DMaxUnpooling.cu154 int64_t numBatch = 1; in max_unpooling2d_forward_out_cuda() local