| /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()
58 float16_t *restrict dscale) { in backwardP1Fp16()
|
| /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) {
61 int ch, float *dxhat_sum, float *dxhathat_sum, float *dbias, float *dscale) {
|
| /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()
121 S *dscale, S *dbias) { in SyncBatchNormGradPostScaleBias()
140 … G *saved_mean, G *saved_invstd, float *dy_sum_red, float *dot_p_red, S *scale, S *dscale, in CalSyncBatchNormGradPost()
|
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | batch_norm_grad.cc | 62 auto dscale = input_args[kInputIndex2]->Broaden(); in BatchNormGradInfer() local
|
| /third_party/mesa3d/src/gallium/auxiliary/gallivm/ |
| D | lp_bld_const.c | 115 double dscale; in lp_const_scale() local
312 double dscale = lp_const_scale(type); in lp_build_const_elem() local
|
| /third_party/flutter/skia/src/core/ |
| D | SkPoint.cpp | 48 double dscale = sk_ieee_double_divide(length, dmag); in set_point_length() local
|
| /third_party/skia/src/core/ |
| D | SkPoint.cpp | 48 double dscale = sk_ieee_double_divide(length, dmag); in set_point_length() local
|
| D | SkPath.cpp | 3655 double dscale = sk_ieee_double_divide(1.0, dmag); in normalize() local
|
| /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
|
| /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
|
| /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
|
| /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
|
| /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
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | prim_nn.cc | 196 auto dscale = args_spec_list[dscale_index]->Broaden(); in InferImplFusedSparseAdam() local
|