| /third_party/mindspore/tests/ut/cpp/session/ |
| D | anf_runtime_algorithm_test.cc | 45 ValueNodePtr value_node = NewValueNode(prim::kPrimAdd); in TEST_F()
64 std::vector<AnfNodePtr> add_inputs{NewValueNode(prim::kPrimAdd)}; in TEST_F()
101 PrimitivePtr add_primitive = prim::kPrimAdd; in TEST_F()
112 std::vector<AnfNodePtr> inputs{NewValueNode(prim::kPrimAdd)}; in TEST_F()
114 EXPECT_EQ(AnfAlgo::GetCNodeName(add), prim::kPrimAdd->name()); in TEST_F()
124 std::vector<AnfNodePtr> inputs{NewValueNode(prim::kPrimAdd)}; in TEST_F()
133 std::vector<AnfNodePtr> inputs{NewValueNode(prim::kPrimAdd)}; in TEST_F()
147 std::vector<AnfNodePtr> add_inputs{NewValueNode(prim::kPrimAdd)}; in TEST_F()
171 std::vector<AnfNodePtr> add_inputs{NewValueNode(prim::kPrimAdd)}; in TEST_F()
195 std::vector<AnfNodePtr> add_inputs{NewValueNode(prim::kPrimAdd)}; in TEST_F()
[all …]
|
| D | kernel_graph_test.cc | 99 auto add_value = NewValueNode(prim::kPrimAdd); in TEST_F()
103 EXPECT_EQ(AnfAlgo::GetCNodeName(new_cnode), prim::kPrimAdd->name()); in TEST_F()
152 std::vector<AnfNodePtr> add_inputs = {NewValueNode(prim::kPrimAdd), x_parameter, y_parameter}; in TEST_F()
177 EXPECT_EQ(AnfAlgo::GetCNodeName(execution_order[0]), prim::kPrimAdd->name()); in TEST_F()
183 EXPECT_EQ(AnfAlgo::GetCNodeName(execution_order[0]), prim::kPrimAdd->name()); in TEST_F()
|
| D | session_basic_test.cc | 55 …std::vector<AnfNodePtr> add_inputs = {NewValueNode(prim::kPrimAdd), original_x_parameter, original… in TEST_F()
90 EXPECT_EQ(AnfAlgo::GetCNodeName(execution_order[0]), prim::kPrimAdd->name()); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/ |
| D | lamb_next_mv_with_decay_rule.cc | 126 VectorRef add2 = VectorRef({prim::kPrimAdd, constant_add2_y_, real_div1}); in DefineAnotherPattern()
129 VectorRef add3 = VectorRef({prim::kPrimAdd, mul4, real_div2}); in DefineAnotherPattern()
143 VectorRef add4 = VectorRef({prim::kPrimAdd, sqrt1, constant_add2_y_}); in DefinePattern()
150 VectorRef add5 = VectorRef({prim::kPrimAdd, mul4, real_div4}); in DefinePattern()
167 VectorRef add2 = VectorRef({prim::kPrimAdd, constant_add2_y_, real_div1}); in DefineAnotherPattern()
170 VectorRef add3 = VectorRef({prim::kPrimAdd, mul4, real_div2}); in DefineAnotherPattern()
184 VectorRef add4 = VectorRef({prim::kPrimAdd, constant_add2_y_, sqrt1}); in DefinePattern()
191 VectorRef add5 = VectorRef({prim::kPrimAdd, mul4, real_div4}); in DefinePattern()
208 VectorRef add2 = VectorRef({prim::kPrimAdd, real_div1, constant_add2_y_}); in DefineAnotherPattern()
211 VectorRef add3 = VectorRef({prim::kPrimAdd, mul4, real_div2}); in DefineAnotherPattern()
[all …]
|
| D | adam_apply_one_with_decay_rule.cc | 58 VectorRef add2({prim::kPrimAdd, add2_y_, sqrt0}); in DefinePattern()
61 VectorRef add3({prim::kPrimAdd, mul4, real_div0}); in DefinePattern()
78 VectorRef add2({prim::kPrimAdd, sqrt0, add2_y_}); in DefinePattern()
81 VectorRef add3({prim::kPrimAdd, mul4, real_div0}); in DefinePattern()
98 VectorRef add2({prim::kPrimAdd, sqrt0, add2_y_}); in DefinePattern()
101 VectorRef add3({prim::kPrimAdd, mul4, real_div0}); in DefinePattern()
118 VectorRef add2({prim::kPrimAdd, add2_y_, sqrt0}); in DefinePattern()
121 VectorRef add3({prim::kPrimAdd, mul4, real_div0}); in DefinePattern()
138 VectorRef add2({prim::kPrimAdd, sqrt0, add2_y_}); in DefinePattern()
141 VectorRef add3({prim::kPrimAdd, mul4, real_div0}); in DefinePattern()
[all …]
|
| D | lamb_next_mv_rule.cc | 133 auto add2 = VectorRef({prim::kPrimAdd, add2_y_, real_div1}); in DefinePattern()
137 return VectorRef({prim::kPrimAdd, mul4, real_div2}); in DefinePattern()
150 VectorRef add4 = VectorRef({prim::kPrimAdd, add2_y_, sqrt1}); in DefineAnotherPattern()
169 auto add2 = VectorRef({prim::kPrimAdd, add2_y_, real_div1}); in DefinePattern()
173 return VectorRef({prim::kPrimAdd, mul4, real_div2}); in DefinePattern()
186 VectorRef add4 = VectorRef({prim::kPrimAdd, sqrt1, add2_y_}); in DefineAnotherPattern()
205 auto add2 = VectorRef({prim::kPrimAdd, real_div1, add2_y_}); in DefinePattern()
209 return VectorRef({prim::kPrimAdd, mul4, real_div2}); in DefinePattern()
222 VectorRef add4 = VectorRef({prim::kPrimAdd, sqrt1, add2_y_}); in DefineAnotherPattern()
241 auto add2 = VectorRef({prim::kPrimAdd, real_div1, add2_y_}); in DefinePattern()
[all …]
|
| D | lamb_next_mv_with_decay_v1_rule.cc | 57 …if (AnfAlgo::GetCNodeName(add5) != prim::kPrimAdd->name() || AnfAlgo::GetInputTensorNum(add5) != k… in MatchAdd5Pattern()
74 …if (AnfAlgo::GetCNodeName(add4) != prim::kPrimAdd->name() || AnfAlgo::GetInputTensorNum(add4) != k… in MatchAdd5Pattern()
127 VectorRef add1({prim::kPrimAdd, mul2, mul3}); in DefinePattern()
129 VectorRef add2({prim::kPrimAdd, real_div1, add2_y_}); in DefinePattern()
133 VectorRef add0({prim::kPrimAdd, mul0, mul1}); in DefinePattern()
137 VectorRef add3({prim::kPrimAdd, real_div2, mul4}); in DefinePattern()
|
| D | adam_apply_one_fusion.cc | 32 …VectorRef true_div0 = VectorRef({prim_real_div, add0, VectorRef({prim::kPrimAdd, sqrt0, add2_y_})}… in DefinePattern()
47 …VectorRef true_div0 = VectorRef({prim_real_div, add0, VectorRef({prim::kPrimAdd, add2_y_, sqrt0})}… in DefinePattern()
62 …VectorRef true_div0 = VectorRef({prim_real_div, add0, VectorRef({prim::kPrimAdd, sqrt0, add2_y_})}… in DefinePattern()
77 …VectorRef true_div0 = VectorRef({prim_real_div, add0, VectorRef({prim::kPrimAdd, sqrt0, add2_y_})}… in DefinePattern()
92 …VectorRef true_div0 = VectorRef({prim_real_div, add0, VectorRef({prim::kPrimAdd, add2_y_, sqrt0})}… in DefinePattern()
108 …VectorRef true_div0 = VectorRef({prim_real_div, add0, VectorRef({prim::kPrimAdd, sqrt0, add2_y_})}… in DefinePattern()
130 …VectorRef true_div0 = VectorRef({prim_real_div, add0, VectorRef({prim::kPrimAdd, add2_y_, sqrt0})}… in DefinePattern()
152 …VectorRef true_div0 = VectorRef({prim_real_div, add0, VectorRef({prim::kPrimAdd, sqrt0, add2_y_})}… in DefinePattern()
174 …VectorRef true_div0 = VectorRef({prim_real_div, add0, VectorRef({prim::kPrimAdd, sqrt0, add2_y_})}… in DefinePattern()
196 …VectorRef true_div0 = VectorRef({prim_real_div, add0, VectorRef({prim::kPrimAdd, add2_y_, sqrt0})}… in DefinePattern()
|
| D | lamb_next_mv_with_decay_rule.h | 41 add0_var_ = std::make_shared<Var>(std::make_shared<Primitive>(prim::kPrimAdd->name())); in MultipleOutputPatternProcessPass()
42 add1_var_ = std::make_shared<Var>(std::make_shared<Primitive>(prim::kPrimAdd->name())); in MultipleOutputPatternProcessPass()
|
| D | adam_apply_one_fusion.h | 41 add0_var_ = std::make_shared<Var>(std::make_shared<Primitive>(prim::kPrimAdd->name())); in PatternProcessPass()
42 add1_var_ = std::make_shared<Var>(std::make_shared<Primitive>(prim::kPrimAdd->name())); in PatternProcessPass()
|
| D | lamb_next_mv_rule.h | 52 add0_var_ = std::make_shared<Var>(std::make_shared<Primitive>(prim::kPrimAdd->name())); in MultipleOutputPatternProcessPass()
53 add1_var_ = std::make_shared<Var>(std::make_shared<Primitive>(prim::kPrimAdd->name())); in MultipleOutputPatternProcessPass()
|
| D | adam_apply_one_with_decay_rule.h | 41 add0_var_ = std::make_shared<Var>(std::make_shared<Primitive>(prim::kPrimAdd->name())); in PatternProcessPass()
42 add1_var_ = std::make_shared<Var>(std::make_shared<Primitive>(prim::kPrimAdd->name())); in PatternProcessPass()
|
| D | lamb_next_right_rule.h | 35 add1_var_(std::make_shared<Var>(std::make_shared<Primitive>(prim::kPrimAdd->name()))) {}
|
| D | prelu_fusion.cc | 31 VectorRef pattern({prim::kPrimAdd, VectorRef({prim::kPrimRelu, x_}), mul_pattern}); in DefinePattern()
|
| D | mul_add_fusion.cc | 62 VectorRef pattern({prim::kPrimAdd, x, y}); in DefinePattern()
|
| D | lamb_next_right_rule.cc | 57 …{prim::kPrimAdd, VectorRef({prim_sqrt, VectorRef({prim::kPrimMul, add1, true_div1_recip_})}), add2… in DefinePattern()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | adam_weight_decay_fusion.cc | 95 VectorRef next_m = VectorRef({prim::kPrimAdd, VectorRef({prim::kPrimMul, beta1_, load_m}), in DefinePattern()
98 VectorRef({prim::kPrimAdd, VectorRef({prim::kPrimMul, beta2_, load_v}), in DefinePattern()
102 …VectorRef({prim::kPrimRealDiv, next_m, VectorRef({prim::kPrimAdd, eps_, VectorRef({prim::kPrimSqrt… in DefinePattern()
103 …VectorRef new_update = VectorRef({prim::kPrimAdd, VectorRef({prim::kPrimMul, weight_decay_, load_p… in DefinePattern()
|
| D | adam_fusion.cc | 95 VectorRef next_m = VectorRef({prim::kPrimAdd, VectorRef({prim::kPrimMul, beta1_, load_m}), in DefinePattern()
98 VectorRef({prim::kPrimAdd, VectorRef({prim::kPrimMul, beta2_, load_v}), in DefinePattern()
102 …VectorRef({prim::kPrimRealDiv, next_m, VectorRef({prim::kPrimAdd, eps_, VectorRef({prim::kPrimSqrt… in DefinePattern()
|
| D | replace_addn_fusion.cc | 38 auto prim = std::make_shared<Primitive>(prim::kPrimAdd->name()); in Process()
|
| D | add_relu_v2_fusion.cc | 56 VectorRef relu = VectorRef({prim::kPrimReluV2, VectorRef({prim::kPrimAdd, x1_, x2_})}); in DefinePattern()
|
| D | add_relu_grad_v2_fusion.cc | 56 …VectorRef relu_grad = VectorRef({prim::kPrimReluGradV2, VectorRef({prim::kPrimAdd, x1_, x2_}), mas… in DefinePattern()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/ |
| D | bnupdate_eltwise_eltwise_fusion_pass.cc | 74 … if (eltwise_input->isa<CNode>() && AnfAlgo::CheckPrimitiveType(eltwise_input, prim::kPrimAdd)) { in MatchSingleFusionPattern()
|
| /third_party/mindspore/tests/ut/cpp/pre_activate/mem_reuse/ |
| D | mem_reuse_test.cc | 165 …std::vector<AnfNodePtr> add_inputs = {NewValueNode(prim::kPrimAdd), original_x_parameter, original… in CreateGraphWithExecOrder()
200 EXPECT_EQ(AnfAlgo::GetCNodeName(execution_order[0]), prim::kPrimAdd->name()); in CreateGraphWithExecOrder()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ |
| D | irpass.cc | 59 … {prim::kPrimScalarAdd, prim::kPrimScalarMul, prim::kPrimAdd, in OptimizeIRPassLib()
178 … {prim::kPrimAdd, prim::kPrimRelu6, prim::kPrimMatMul, prim::kPrimMakeTuple, prim::kPrimMaxPool}); in OptimizeIRPassLib()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | less_batch_normalization.cc | 139 {prim::kPrimTupleGetItem, prim::kPrimBatchNorm, prim::kPrimConv2D, prim::kPrimAdd},
149 …{kFirstBranchPattern9, {prim::kPrimAdd}, {kFirstBranchStartIndexPattern9, kFirstBranchEndIndexPatt…
153 prim::kPrimConv2D, prim::kPrimAdd},
|