| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | sync_batch_norm_grad_impl.cu | 98 S *dscale, S *dbias, float epsilon) { in SyncBatchNormGradPost() argument
121 S *dscale, S *dbias) { in SyncBatchNormGradPostScaleBias() argument
126 dscale[plane] = static_cast<S>(dot_p_plane * invstd); in SyncBatchNormGradPostScaleBias()
140 … G *saved_mean, G *saved_invstd, float *dy_sum_red, float *dot_p_red, S *scale, S *dscale, in CalSyncBatchNormGradPost() argument
143 … dy_sum_red, dot_p_red, scale, dscale, dbias, epsilon); in CalSyncBatchNormGradPost()
145 C, saved_invstd, dy_sum_red, dot_p_red, dscale, dbias); in CalSyncBatchNormGradPost()
164 … float *dot_p_red, float *scale, float *dscale, float *dbias,
169 … float *scale, float *dscale, float *dbias, float epsilon,
174 … half *scale, half *dscale, half *dbias, float epsilon,
179 … half *scale, half *dscale, half *dbias, float epsilon,
[all …]
|
| D | sync_batch_norm_grad_impl.cuh | 25 … G *saved_mean, G *invstd_saved, float *dy_sum_red, float *dot_p_red, S *scale, S *dscale,
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32_grad/ |
| D | batch_norm.c | 27 … int ch, float *dxhat_sum, float *dxhathat_sum, float *dbias, float *dscale, float *dx) { argument
31 … float *restrict dxhathat_sum, float *restrict dbias, float *restrict dscale, float *restrict dx) {
41 dscale[c] += (yt[ix] * x_hat);
61 int ch, float *dxhat_sum, float *dxhathat_sum, float *dbias, float *dscale) { argument
65 float *restrict dxhathat_sum, float *restrict dbias, float *restrict dscale) {
73 dscale[c] += (yt[ix] * x_hat);
|
| D | batch_norm.h | 33 … int ch, float *dxhat_sum, float *dxhathat_sum, float *dbias, float *dscale, float *dx);
35 int ch, float *dxhat_sum, float *dxhathat_sum, float *dbias, float *dscale);
|
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | batch_norm_grad.cc | 62 auto dscale = input_args[kInputIndex2]->Broaden(); in BatchNormGradInfer() local
66 AbstractBasePtrList rets = {dx, dscale, dscale, reserve_1, reserve_2}; in BatchNormGradInfer()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16_grad/ |
| D | batch_norm.c | 29 float16_t *restrict dscale, float16_t *restrict dx) { in backwardAllFp16() argument
37 dscale[c] += (yt[ix] * x_hat); in backwardAllFp16()
58 float16_t *restrict dscale) { in backwardP1Fp16() argument
65 dscale[c] += (yt[ix] * x_hat); in backwardP1Fp16()
|
| D | batch_norm.h | 29 float16_t *dscale, float16_t *dx);
32 float16_t *dscale);
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16_grad/ |
| D | bn_fp16_grad.cc | 111 float16_t *dscale = reinterpret_cast<float16_t *>(output_scale->data()); in DoExecute() local
130 std::fill(dscale, dscale + channels, 0.f); in DoExecute()
135 …16(x, yt, save_mean, save_var, scale, total, channels, dxhat_sum, dxhathat_sum, dbias, dscale, dx); in DoExecute()
140 …P1Fp16(x, yt, save_mean, save_var, scale, total, channels, dxhat_sum, dxhathat_sum, dbias, dscale); in DoExecute()
|
| /third_party/mesa3d/src/gallium/auxiliary/gallivm/ |
| D | lp_bld_const.c | 115 double dscale; in lp_const_scale() local
119 dscale = (double)llscale; in lp_const_scale()
120 assert((unsigned long long)dscale == llscale); in lp_const_scale()
122 return dscale; in lp_const_scale()
312 double dscale = lp_const_scale(type); in lp_build_const_elem() local
314 elem = LLVMConstInt(elem_type, (long long) round(val*dscale), 0); in lp_build_const_elem()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32_grad/ |
| D | bn_grad.cc | 116 float *dscale = reinterpret_cast<float *>(output_scale->MutableData()); in Execute() local
143 std::fill(dscale, dscale + channels, 0.f); in Execute()
148 …ll(x, yt, save_mean, save_var, scale, total, channels, dxhat_sum, dxhathat_sum, dbias, dscale, dx); in Execute()
153 …wardP1(x, yt, save_mean, save_var, scale, total, channels, dxhat_sum, dxhathat_sum, dbias, dscale); in Execute()
|
| /third_party/flutter/skia/src/core/ |
| D | SkPoint.cpp | 48 double dscale = sk_ieee_double_divide(length, dmag); in set_point_length() local
49 x *= dscale; in set_point_length()
50 y *= dscale; in set_point_length()
|
| /third_party/skia/src/core/ |
| D | SkPoint.cpp | 48 double dscale = sk_ieee_double_divide(length, dmag); in set_point_length() local
49 x *= dscale; in set_point_length()
50 y *= dscale; in set_point_length()
|
| D | SkPath.cpp | 3655 double dscale = sk_ieee_double_divide(1.0, dmag); in normalize() local
3656 a *= dscale; in normalize()
3657 b *= dscale; in normalize()
3658 c *= dscale; in normalize()
|
| /third_party/mindspore/mindspore/ops/_grad/ |
| D | grad_other_ops.py | 64 dscale = out[1]
66 return dx, dscale, dbias, zeros_like(mean), zeros_like(variance)
|
| D | grad_nn_ops.py | 719 dscale = out[1]
721 return dx, dscale, dbias, zeros_like(mean), zeros_like(variance)
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32_grad/ |
| D | bn_grad_fp32_test.cc | 101 auto dscale = reinterpret_cast<float *>(outputs[1]->MutableData()); in TEST_F() local
102 for (int i = 0; i < channels; i++) std::cout << dscale[i] << " "; in TEST_F()
104 res = CompareRelativeOutput(dscale, "./bngrad/output_dscale_3.bin"); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
| D | batch_norm_grad_gpu_kernel.h | 86 auto dscale = GetDeviceAddress<float>(outputs, 1); in Launch() local
105 … dz_desc_, dz, dx_desc_, dx, scale_bias_diff_desc_, scale, bias, dscale, dbias, in Launch()
110 …CalBatchNormGrad(x, dy, scale, save_mean, save_variance, dx, dscale, dbias, epsilon_, batch_, chan… in Launch()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nccl/ |
| D | sync_batch_norm_grad_gpu_kernel.h | 54 S *dscale = GetDeviceAddress<S>(outputs, 1); in Launch() local
64 dscale, dbias, epsilon_, reinterpret_cast<cudaStream_t>(stream_ptr)); in Launch()
|
| /third_party/mindspore/tests/vm_impl/ |
| D | nn_ops_vm_impl.py | 216 dx, dscale, dshift = vm.batch_norm_grad(dy, x, scale, save_mean, save_inv_variance)
217 return (Tensor(dx), Tensor(dscale), Tensor(dshift))
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | prim_nn.cc | 196 auto dscale = args_spec_list[dscale_index]->Broaden(); in InferImplFusedSparseAdam() local
199 AbstractBasePtrList rets = {dx, dscale, dbias}; in InferImplFusedSparseAdam()
|