Searched defs:layer_norm (Results 1 – 13 of 13) sorted by relevance
54 inline Tensor layer_norm( in layer_norm() function77 inline Tensor layer_norm( in layer_norm() function
39 def layer_norm( function
2763 def layer_norm( function
81 static Tensor layer_norm( in layer_norm() function
53 def layer_norm( function
92 def layer_norm( function
24 void layer_norm( in layer_norm() function
27 void layer_norm( in layer_norm() function
39 def layer_norm(self, data, shape): member in TorchTensorEngine
145 def layer_norm(x, weight, bias): function
915 auto layer_norm = getObservedQParamOpFusionInfo( in quant_fusion_pattern_and_replacements() local
346 def layer_norm( function
2879 def layer_norm( function