Home
last modified time | relevance | path

Searched defs:max_pool2d_with_indices (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DDilatedMaxPool2d.cpp18 TORCH_META_FUNC(max_pool2d_with_indices) in TORCH_META_FUNC() argument
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/
H A Dpooling.h262 inline std::tuple<Tensor, Tensor> max_pool2d_with_indices( in max_pool2d_with_indices() function
283 inline std::tuple<Tensor, Tensor> max_pool2d_with_indices( in max_pool2d_with_indices() function
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Ddecomposition.py933 def max_pool2d_with_indices( function
/aosp_15_r20/external/pytorch/torch/jit/
H A D_shape_functions.py346 def max_pool2d_with_indices( function
/aosp_15_r20/external/pytorch/torch/nn/
H A Dfunctional.py754 def max_pool2d_with_indices( function
/aosp_15_r20/external/pytorch/torch/onnx/
H A Dsymbolic_opset9.py1499 max_pool2d_with_indices = _onnx_symbolic("aten::max_pool2d_with_indices")( variable