Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/
H A Dpooling.h599 inline Tensor adaptive_avg_pool3d( in adaptive_avg_pool3d() function
622 inline Tensor adaptive_avg_pool3d( in adaptive_avg_pool3d() function
/aosp_15_r20/external/pytorch/torch/ao/nn/quantized/
H A Dfunctional.py153 def adaptive_avg_pool3d(input: Tensor, output_size: BroadcastingList2[int]) -> Tensor: function
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/
H A Dquantization_patterns.h833 auto adaptive_avg_pool3d = getInputTensorQParamOpFusionInfo( in quant_fusion_pattern_and_replacements() local
/aosp_15_r20/external/pytorch/torch/nn/
H A Dfunctional.py1385 def adaptive_avg_pool3d(input: Tensor, output_size: BroadcastingList3[int]) -> Tensor: function