Searched defs:relu_grad (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | batch_norm_add_relu_grad_fusion.cc | 68 …AndType(const CNodePtr &bn_add_relu_grad, const AnfNodePtr &bn_grad, const AnfNodePtr &relu_grad) { in SetShapeAndType()
83 …id ReplaceOutput(const FuncGraphPtr &graph, const AnfNodePtr &bn_grad, const AnfNodePtr &relu_grad, in ReplaceOutput()
129 auto relu_grad = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(node), 0); in PatternCheck() local
152 VectorRef relu_grad = VectorRef({prim::kPrimReluGrad, dy_, y_}); in DefinePattern() local
167 auto relu_grad = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(node), 0); in Process() local
|
| D | relu_v2_pass.cc | 34 CNodePtr GetRelu(const CNodePtr &relu_grad) { in GetRelu()
95 CNodePtr CreateReluGradV2(const FuncGraphPtr &graph, const CNodePtr &relu_grad, const AnfNodePtr &s… in CreateReluGradV2()
126 VectorRef relu_grad({prim::kPrimReluGrad, dy_, VectorRef({prim::kPrimRelu, x_})}); in DefinePattern() local
133 auto relu_grad = node->cast<CNodePtr>(); in Process() local
|
| D | batch_norm_relu_grad_fusion.cc | 32 VectorRef relu_grad = VectorRef({prim::kPrimReluGrad, dy_, y_}); in DefinePattern() local
63 auto relu_grad = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(node), 0); in Process() local
|
| D | add_relu_grad_v2_fusion.cc | 56 …VectorRef relu_grad = VectorRef({prim::kPrimReluGradV2, VectorRef({prim::kPrimAdd, x1_, x2_}), mas… in DefinePattern() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/ |
| D | derelu_fusion.cc | 32 CNodePtr GetRelu(const CNodePtr &relu_grad) { in GetRelu()
83 CNodePtr CreateReluGradV2(const FuncGraphPtr &graph, const CNodePtr &relu_grad, const AnfNodePtr &s… in CreateReluGradV2()
102 VectorRef relu_grad({prim::kPrimReluGrad, i0, relu}); in DefinePattern() local
110 auto relu_grad = node->cast<CNodePtr>(); in Process() local
|
| /third_party/mindspore/tests/ut/cpp/python_input/gtest_input/pre_activate/ |
| D | derelu_fusion.py | 20 relu_grad = Primitive('ReluGrad') variable
|
| /third_party/mindspore/tests/vm_impl/ |
| D | vm_me.py | 493 def relu_grad(y): function
|