Searched refs:batch_norm_grad (Results 1 – 9 of 9) sorted by relevance
| /third_party/mindspore/tests/ut/cpp/ops/ |
| D | test_ops_batch_norm_grad.cc | 36 auto batch_norm_grad = std::make_shared<BatchNormGrad>(); in TEST_F() local
37 batch_norm_grad->Init(); in TEST_F()
38 EXPECT_EQ(batch_norm_grad->get_is_training(), false); in TEST_F()
39 EXPECT_EQ((int64_t)(batch_norm_grad->get_epsilon() - 1e-05), 0); in TEST_F()
45 auto abstract = batch_norm_grad->Infer( in TEST_F()
|
| /third_party/mindspore/tests/ut/cpp/python_input/gtest_input/pre_activate/ |
| D | batchnormgrad_to_bninfergrad.py | 20 batch_norm_grad = G.BatchNormGrad(is_training=False) variable
42 res = batch_norm_grad(input0, input1, input2, input3, input4, input5)
53 res = batch_norm_grad(input0, input1, input2, input3, input4, input5)
|
| D | batch_norm_grad_split.py | 20 batch_norm_grad = G.BatchNormGrad(is_training=True) variable
43 bn_grad_output = batch_norm_grad(i0, i1, i2, i3, i4, i5)
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | batch_norm_relu_grad_fusion.cc | 33 VectorRef batch_norm_grad = in DefinePattern() local
35 return batch_norm_grad; in DefinePattern()
|
| D | batch_norm_add_relu_grad_fusion.cc | 153 VectorRef batch_norm_grad = in DefinePattern() local
155 return batch_norm_grad; in DefinePattern()
|
| /third_party/mindspore/tests/vm_impl/ |
| D | vm_interface.py | 28 setattr(vm, 'batch_norm_grad', batch_norm_grad)
|
| D | nn_ops_vm_impl.py | 216 dx, dscale, dshift = vm.batch_norm_grad(dy, x, scale, save_mean, save_inv_variance)
|
| D | vm_me.py | 143 def batch_norm_grad(dy, x, scale, save_mean, save_inv_variance): function
|
| /third_party/mindspore/config/ |
| D | op_info.config | 186 …, "async_flag": false, "binfile_name": "batch_norm_grad.so", "compute_cost": 10, "kernel_name": "b…
|