Home
last modified time | relevance | path

Searched defs:group_norm (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/
H A Dnormalization.h169 inline Tensor group_norm( in group_norm() function
198 inline Tensor group_norm( in group_norm() function
/aosp_15_r20/external/executorch/kernels/portable/cpu/
H A Dop_native_group_norm.cpp24 void group_norm( in group_norm() function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A Dgroup_norm.cpp177 Tensor group_norm( in group_norm() function
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/
H A Dquantization_patterns.h921 auto group_norm = getObservedQParamOpFusionInfo( in quant_fusion_pattern_and_replacements() local
/aosp_15_r20/external/pytorch/torch/_refs/nn/functional/
H A D__init__.py306 def group_norm( function
/aosp_15_r20/external/pytorch/torch/nn/
H A Dfunctional.py2922 def group_norm( function
/aosp_15_r20/external/pytorch/torch/onnx/
H A Dsymbolic_opset9.py5713 def group_norm( function