Home
last modified time | relevance | path

Searched defs:avg_pool1d (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DPooling.cpp114 Tensor avg_pool1d( in avg_pool1d() function
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/
H A Dpooling.h14 inline Tensor avg_pool1d( in avg_pool1d() function
39 inline Tensor avg_pool1d( in avg_pool1d() function
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/
H A Dquantization_patterns.h794 auto avg_pool1d = getInputTensorQParamOpFusionInfo( in quant_fusion_pattern_and_replacements() local
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dnn_ops.py4620 def avg_pool1d(input, ksize, strides, padding, data_format="NWC", name=None): # pylint: disable=re… function
/aosp_15_r20/external/pytorch/torch/nn/
H A Dfunctional.py345 avg_pool1d = _add_docstr( variable