Home
last modified time | relevance | path

Searched defs:avg_pool3d (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/
H A Dpooling.h100 inline Tensor avg_pool3d( in avg_pool3d() function
132 inline Tensor avg_pool3d( in avg_pool3d() function
/aosp_15_r20/external/pytorch/torch/ao/nn/quantized/
H A Dfunctional.py89 def avg_pool3d( function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DAveragePool3d.cpp20 TORCH_META_FUNC(avg_pool3d) ( in TORCH_META_FUNC() argument
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/
H A Dquantization_patterns.h818 auto avg_pool3d = getInputTensorQParamOpFusionInfo( in quant_fusion_pattern_and_replacements() local
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dnn_ops.py4667 def avg_pool3d(input, ksize, strides, padding, data_format="NDHWC", name=None): # pylint: disable=… function
/aosp_15_r20/external/pytorch/torch/nn/
H A Dfunctional.py407 avg_pool3d = _add_docstr( variable
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dmkl_layout_pass.cc857 string avg_pool3d; member
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dlowering.py4557 def avg_pool3d( function