| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | fused_batchnorm_fp32.cc | 83 float *save_mean = static_cast<float *>(in_tensors_.at(FOURTH_INPUT)->data()); in Run() local
120 float *save_mean = static_cast<float *>(in_tensors_.at(FOURTH_INPUT)->data()); in Eval() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
| D | fused_batchnorm_fp16.cc | 38 …p16CPUKernel::CalcMeanVar(float16_t *in, float16_t *scale, float16_t *offset, float16_t *save_mean, in CalcMeanVar()
146 float16_t *save_mean = static_cast<float16_t *>(in_tensors_.at(kInCurrentMeanIdx)->data()); in Eval() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | batchnorm_grad_impl.cu | 30 __global__ void BatchNormGradKernel(T *x_input, T *dy, float *scale, float *save_mean, float *save_… in BatchNormGradKernel()
108 void CalBatchNormGrad(T *x, T *dy, float *scale, float *save_mean, float *save_variance, T *dx, flo… in CalBatchNormGrad()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | batch_norm_relu_grad_fusion.cc | 80 auto save_mean = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(node), 3); in Process() local
|
| D | batch_norm_add_relu_grad_fusion.cc | 177 auto save_mean = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(node), 3); in Process() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | batchnorm_fp32.c | 63 float *save_mean, float *save_var) { in FusedBatchNormFp32MeanVar()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | batchnorm_fp16.c | 62 … const BatchNormParameter *param, float16_t *save_mean, float16_t *save_var) { in FusedBatchNormFp16MeanVar()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32_grad/ |
| D | bn_grad.cc | 90 float *save_mean = reinterpret_cast<float *>(input_mean->MutableData()); in Execute() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16_grad/ |
| D | bn_fp16_grad.cc | 90 float16_t *save_mean = reinterpret_cast<float16_t *>(input_mean->data()); in DoExecute() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
| D | batch_norm_gpu_kernel.h | 65 auto save_mean = GetDeviceAddress<float>(outputs, 3); in Launch() local
|
| D | instance_norm_grad_gpu_kernel.h | 66 auto save_mean = GetDeviceAddress<float>(inputs, 3); in Launch() local
|
| D | batch_norm_grad_gpu_kernel.h | 76 auto save_mean = GetDeviceAddress<float>(inputs, 3); in Launch() local
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32_grad/ |
| D | bn_grad_fp32_test.cc | 184 float *save_mean = reinterpret_cast<float *>(save_mean_tensor.MutableData()); in TEST_F() local
|
| /third_party/mindspore/tests/vm_impl/ |
| D | vm_me.py | 121 def _batch_norm_grad(dout, x, scale, save_mean, save_inv_variance, \ argument
143 def batch_norm_grad(dy, x, scale, save_mean, save_inv_variance): argument
|
| D | nn_ops_vm_impl.py | 210 def vm_impl(dy, x, scale, save_mean, save_inv_variance): argument
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | _grad_ops.py | 689 def __call__(self, dy, x, scale, save_mean, save_inv_variance): argument
|