Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DPooling.cpp43 Tensor adaptive_avg_pool1d(const Tensor & self, IntArrayRef output_size) { in adaptive_avg_pool1d() function
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/
H A Dpooling.h539 inline Tensor adaptive_avg_pool1d( in adaptive_avg_pool1d() function
560 inline Tensor adaptive_avg_pool1d( in adaptive_avg_pool1d() function
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/
H A Dquantization_patterns.h827 auto adaptive_avg_pool1d = getInputTensorQParamOpFusionInfo( in quant_fusion_pattern_and_replacements() local
/aosp_15_r20/external/pytorch/torch/nn/
H A Dfunctional.py1354 adaptive_avg_pool1d = _add_docstr( variable