Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A DAdaptiveAveragePooling.cpp63 int64_t isizeD, // Set to 1 for 2D in adaptive_avg_pool_single_out_frame()
187 int64_t isizeD = kSpatialDim == 2 ? 1 : input.size(-3); in _adaptive_avg_pool() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/
H A DQuantizedOpKernels.cpp1818 int64_t isizeD, // Set to 1 for 2d
1962 int64_t isizeD,