| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | layer_norm_fp32.c | 28 float32x4_t square_sum = vdupq_n_f32(0); in LayerNormMeanAndSquare() local
33 square_sum = vaddq_f32(square_sum, squarev); in LayerNormMeanAndSquare()
36 *square_mean = square_sum[0] + square_sum[1] + square_sum[2] + square_sum[3]; in LayerNormMeanAndSquare()
|
| D | l2_norm_fp32.c | 59 float square_sum = 0.0f; in ThreadTrailingAxis() local
63 square_sum += val * val; in ThreadTrailingAxis()
65 float sqrt_sum = sqrtf(square_sum > param->epsilon_ ? square_sum : param->epsilon_); in ThreadTrailingAxis()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/ |
| D | l2_norm_int8.c | 26 int32_t square_sum = 0.0f; in L2NormalizationInt8() local
29 square_sum += in * in; in L2NormalizationInt8()
33 GetSqrtQuantMultiplierExp(square_sum, -1, &multiplier, &shift); in L2NormalizationInt8()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | layer_norm_fp16.c | 27 float square_sum = 0.0f; in LayerNormMeanAndSquareFp16() local
31 square_sum += srcv[i] * srcv[i]; in LayerNormMeanAndSquareFp16()
39 square_sum += src[index] * src[index]; in LayerNormMeanAndSquareFp16()
42 *square_mean = (float16_t)(square_sum / num); in LayerNormMeanAndSquareFp16()
|
| /third_party/ffmpeg/libavfilter/ |
| D | af_axcorrelate.c | 90 static float square_sum(const float *x, const float *y, int size) in square_sum() function
92 float square_sum = 0.f; in square_sum() local
95 square_sum += x[i] * y[i]; in square_sum()
97 return square_sum; in square_sum()
169 num_sum[0] = square_sum(x, y, size); in xcorrelate_fast()
170 den_sumx[0] = square_sum(x, x, size); in xcorrelate_fast()
171 den_sumy[0] = square_sum(y, y, size); in xcorrelate_fast()
|
| D | vf_yaepblur.c | 191 …uint64_t sum, square_sum, mean, var; …
205 … square_sum = square_sat[higher_y * sat_linesize + higher_x] \
210 … var = (square_sum - sum * sum / count) / count; \
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | layer_norm_cpu_kernel.cc | 98 T square_sum = (T)0.0; in LaunchKernel() local
101 square_sum += x[j] * x[j]; in LaunchKernel()
104 T block_var = square_sum / static_cast<T>(block_size_) - block_mean * block_mean; in LaunchKernel()
|
| /third_party/mindspore/mindspore/ops/composite/ |
| D | clip_ops.py | 128 square_sum = self.hyper_map(get_square_sum, x)
129 global_norm = F.sqrt(F.addn(square_sum))
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/ |
| D | reduce_ops_declare.cc | 24 OUTPUT_MAP(BNTrainingReduce) = {{0, OUTPUT_DESC(sum)}, {1, OUTPUT_DESC(square_sum)}};
36 …NTrainingUpdate) = {{1, INPUT_DESC(x)}, {2, INPUT_DESC(sum)}, {3, INPUT_DESC(square_sum)},
|
| /third_party/ffmpeg/libavcodec/ |
| D | vp6.c | 550 int sum = 0, square_sum = 0; in vp6_block_variance() local
556 square_sum += src[x]*src[x]; in vp6_block_variance()
560 return (16*square_sum - sum*sum) >> 8; in vp6_block_variance()
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | nn_ops.py | 1143 def infer_shape(self, x, sum, square_sum, scale, b, mean, variance): argument
1146 validator.check_equal_int(len(square_sum), 1, "square_sum rank", self.name)
1152 validator.check("square_sum shape", square_sum, "sum", sum, Rel.EQ, self.name)
1159 def infer_dtype(self, x, sum, square_sum, scale, b, mean, variance): argument
1163 (x, sum, square_sum, scale, b, mean, variance)))
|
| /third_party/mindspore/config/ |
| D | op_info.config | 244 …alse, "param_type": "required", "shape": "all"}, {"index": 1, "name": "square_sum", "need_compile"…
246 …alse, "param_type": "required", "shape": "all"}, {"index": 2, "name": "square_sum", "need_compile"…
398 …alse, "param_type": "required", "shape": "all"}, {"index": 2, "name": "square_sum", "need_compile"…
399 …alse, "param_type": "required", "shape": "all"}, {"index": 2, "name": "square_sum", "need_compile"…
|