Home
last modified time | relevance | path

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

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