Home
last modified time | relevance | path

Searched defs:adaptive_max_pool3d (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/
H A Dpooling.h508 inline Tensor adaptive_max_pool3d( in adaptive_max_pool3d() function
529 inline Tensor adaptive_max_pool3d( in adaptive_max_pool3d() function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DAdaptiveMaxPooling3d.cpp18 TORCH_META_FUNC(adaptive_max_pool3d) (const Tensor& input, IntArrayRef output_size) { in TORCH_META_FUNC() argument
/aosp_15_r20/external/pytorch/torch/nn/
H A Dfunctional.py1343 adaptive_max_pool3d = boolean_dispatch( variable