Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DPooling.cpp167 Tensor max_pool3d( in max_pool3d() function
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/
H A Dpooling.h297 inline Tensor max_pool3d( in max_pool3d() function
322 inline Tensor max_pool3d( in max_pool3d() function
/aosp_15_r20/external/pytorch/torch/csrc/lazy/ts_backend/
H A Dts_native_functions.cpp326 at::Tensor LazyNativeFunctions::max_pool3d( in max_pool3d() function in torch::lazy::LazyNativeFunctions
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dnn_ops.py5080 def max_pool3d(input, ksize, strides, padding, data_format="NDHWC", name=None): function
/aosp_15_r20/external/pytorch/torch/nn/
H A Dfunctional.py923 max_pool3d = boolean_dispatch( variable
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dmkl_layout_pass.cc898 string max_pool3d; member