Home
last modified time | relevance | path

Searched defs:fractional_max_pool2d (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/
H A Dpooling.h875 inline Tensor fractional_max_pool2d( in fractional_max_pool2d() function
897 inline Tensor fractional_max_pool2d( in fractional_max_pool2d() function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DFractionalMaxPool2d.cpp20 TORCH_META_FUNC(fractional_max_pool2d) ( in TORCH_META_FUNC() argument
/aosp_15_r20/external/pytorch/torch/nn/
H A Dfunctional.py537 fractional_max_pool2d = boolean_dispatch( variable
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dlowering.py4411 def fractional_max_pool2d(x, kernel_size, output_size, random_samples): function