Searched defs:M_N (Results 1 – 7 of 7) sorted by relevance
| /external/pytorch/aten/src/ATen/native/ |
| D | layer_norm.cpp | 90 auto M_N = _check_layer_norm_inputs(input, normalized_shape, weight, bias); in layer_norm_cpu() local
129 auto M_N = _check_layer_norm_inputs(input, normalized_shape, weight, bias); in layer_norm_backward_cpu() local
215 auto M_N = _check_layer_norm_inputs(input, normalized_shape, weight, bias); in math_native_layer_norm() local
|
| /external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| D | qnormalization.cpp | 32 auto M_N = _check_layer_norm_inputs(input, normalized_shape, weight, bias); in quantized_layer_norm_impl() local
|
| /external/pytorch/aten/src/ATen/native/nested/ |
| D | NestedTensorBackward.cpp | 211 auto M_N = _check_nested_layer_norm_inputs( in layer_norm_backward_nested() local
|
| D | NestedTensorMath.cpp | 161 auto M_N = _check_nested_layer_norm_inputs(*nt_input, normalized_shape, weight, bias); in nested_layer_norm() local
|
| /external/pytorch/aten/src/ATen/native/mkldnn/ |
| D | Normalization.cpp | 98 auto M_N = at::native::_check_layer_norm_inputs(input, normalized_shape, weight, bias); in mkldnn_layer_norm_last_index_weight_bias_f32() local
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | layer_norm_kernel.cu | 1348 auto M_N = _check_layer_norm_inputs(input, normalized_shape, weight, bias); in layer_norm_cuda() local
1404 auto M_N = _check_layer_norm_inputs(input, normalized_shape, weight, bias); in layer_norm_backward_cuda() local
|
| /external/pytorch/torch/csrc/jit/runtime/static/ |
| D | ops.cpp | 2104 auto M_N = at::native::_check_layer_norm_inputs( in __anon5bf8650b6602() local
|