Searched defs:batch_norm (Results 1 – 9 of 9) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | batch_norm_relu_fusion.cc | 31 VectorRef batch_norm = VectorRef({prim::kPrimBatchNorm, x_, scale_, bias_, mean_, var_}); in DefinePattern() local
44 auto batch_norm = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(tuple_get_item), 0); in Process() local
|
| D | batch_norm_add_relu_fusion.cc | 31 VectorRef batch_norm = VectorRef({prim::kPrimBatchNorm, x_, scale_, bias_, mean_, var_}); in DefinePattern() local
47 auto batch_norm = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(tuple_get_item), 0); in Process() local
|
| D | post_batch_norm_add_relu_fusion.cc | 31 VectorRef batch_norm = VectorRef({prim::kPrimBatchNorm, x_, scale_, bias_, mean_, var_}); in DefinePattern() local
47 auto batch_norm = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(tuple_get_item), 0); in Process() local
|
| D | batch_norm_relu_grad_fusion.cc | 86 auto batch_norm = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(save_mean), 0); in Process() local
|
| D | batch_norm_add_relu_grad_fusion.cc | 183 auto batch_norm = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(save_mean), 0); in Process() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/ |
| D | fused_batch_norm_fusion.cc | 249 …VectorRef batch_norm = VectorRef({batch_norm_var_, data_input0_var_, data_input1_var_, data_input2… in DefinePattern() local
268 …VectorRef batch_norm = VectorRef({batch_norm_var_, data_input0_var_, data_input1_var_, data_input2… in DefinePattern() local
291 …VectorRef batch_norm = VectorRef({batch_norm_var_, data_input0_var_, data_input1_var_, data_input2… in DefinePattern() local
|
| /third_party/mindspore/tests/ut/cpp/python_input/gtest_input/pre_activate/ |
| D | batchnorm_to_bninfer.py | 20 batch_norm = P.BatchNorm(is_training=False) variable
|
| /third_party/mindspore/tests/vm_impl/ |
| D | vm_me.py | 107 def batch_norm(x, scale=1, shift=0, mean=None, variance=None, function
|
| /third_party/mindspore/tests/ut/cpp/transform/ |
| D | convert_test.cc | 137 PrimitivePtr batch_norm = prim::kPrimBatchNorm; in TEST_F() local
|