Home
last modified time | relevance | path

Searched refs:computeAdaptiveAvgPool2d (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/operators/
H A Dreduction.h21 TORCH_API Tensor computeAdaptiveAvgPool2d(
H A Dreduction.cpp161 Tensor computeAdaptiveAvgPool2d( in computeAdaptiveAvgPool2d() function
/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Dlowerings.cpp1887 computeAdaptiveAvgPool2d); in nnc_lowerings_lazy_registration()