Home
last modified time | relevance | path

Searched refs:VarPtr (Results 1 – 25 of 150) sorted by relevance

123456

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/
Doptimizer_unify_output.cc65 VarPtr var = std::make_shared<Var>(); in DefinePattern()
66 VarPtr accum = std::make_shared<Var>(); in DefinePattern()
67 VarPtr linear = std::make_shared<Var>(); in DefinePattern()
68 VarPtr grad = std::make_shared<Var>(); in DefinePattern()
69 VarPtr lr = std::make_shared<Var>(); in DefinePattern()
70 VarPtr l1 = std::make_shared<Var>(); in DefinePattern()
71 VarPtr l2 = std::make_shared<Var>(); in DefinePattern()
72 VarPtr lr_power = std::make_shared<Var>(); in DefinePattern()
73 VarPtr u = std::make_shared<SeqVar>(); in DefinePattern()
83 VarPtr var = std::make_shared<Var>(); in DefinePattern()
[all …]
/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Dtflite_rel_pos_multi_head_attention_fusion.h47 … const std::vector<VarPtr> &stack_params, const VarPtr &full_connect_prim,
60 mutable VarPtr query_u_{nullptr};
61 mutable VarPtr query_v_{nullptr};
62 mutable VarPtr query_prim_{nullptr};
63 mutable VarPtr key_prim_{nullptr};
64 mutable VarPtr value_prim_{nullptr};
65 mutable VarPtr pos_prim_{nullptr};
66 mutable VarPtr output_prim_{nullptr};
67 mutable VarPtr input_p_{nullptr};
68 mutable VarPtr weight_p_{nullptr};
[all …]
Dmulti_head_attention_fusion.h65 mutable VarPtr input_q_{nullptr};
66 mutable VarPtr input_k_{nullptr};
67 mutable VarPtr input_v_{nullptr};
69 mutable VarPtr weight_q_{nullptr};
70 mutable VarPtr weight_k_{nullptr};
71 mutable VarPtr weight_v_{nullptr};
72 mutable VarPtr weight_o_{nullptr};
73 mutable VarPtr bias_q_{nullptr};
74 mutable VarPtr bias_k_{nullptr};
75 mutable VarPtr bias_v_{nullptr};
[all …]
Dtf_gelu_fusion.h40 mutable VarPtr power_ = nullptr;
41 mutable VarPtr power_y_ = nullptr;
42 mutable VarPtr mul1_x_ = nullptr;
43 mutable VarPtr mul2_x_ = nullptr;
44 mutable VarPtr tanh_ = nullptr;
45 mutable VarPtr add2_x_ = nullptr;
46 mutable VarPtr mul3_x_ = nullptr;
Dnorm_fusion.h59 mutable VarPtr input_ = nullptr;
60 mutable VarPtr mean1_ = nullptr;
61 mutable VarPtr mean1_axes_ = nullptr;
62 mutable VarPtr mean2_ = nullptr;
63 mutable VarPtr mean2_axes_ = nullptr;
64 mutable VarPtr gamma_ = nullptr;
65 mutable VarPtr beta_ = nullptr;
66 mutable VarPtr epsilon_ = nullptr;
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/
Dlamb_next_mv_with_decay_v1_rule.h51 VarPtr input0_;
52 VarPtr input1_;
53 VarPtr input2_;
54 VarPtr input3_;
55 VarPtr input4_;
56 VarPtr input5_;
57 VarPtr input6_;
58 VarPtr mul0_x_;
59 VarPtr mul1_sub_;
60 VarPtr mul2_x_;
[all …]
Dlamb_next_mv_rule.h67 VarPtr input0_;
68 VarPtr input1_;
69 VarPtr input2_;
70 VarPtr input3_;
71 VarPtr input4_;
72 VarPtr input5_;
73 VarPtr input6_;
74 VarPtr mul0_x_;
75 VarPtr mul1_sub_;
76 VarPtr mul2_x_;
[all …]
Dlamb_update_with_lr_rule_fusion.h43 VarPtr input0_;
44 VarPtr input1_;
45 VarPtr input2_;
46 VarPtr input3_;
47 VarPtr input4_;
48 VarPtr input5_;
49 VarPtr constant_greater_max_;
50 VarPtr constant_select_;
51 VarPtr constant_minimum_;
Dfused_batch_norm_fusion.h60 VarPtr data_input0_var_;
61 VarPtr data_input1_var_;
62 VarPtr data_input2_var_;
63 VarPtr variable_input0_var_;
64 VarPtr variable_input1_var_;
65 VarPtr constant_input0_var_;
66 VarPtr constant_input1_var_;
67 VarPtr batch_norm_var_;
68 VarPtr assign_sub0_var_;
69 VarPtr assign_sub1_var_;
[all …]
Dadam_apply_one_with_decay_rule.h51 VarPtr input0_;
52 VarPtr input1_;
53 VarPtr input2_;
54 VarPtr input3_;
55 VarPtr input4_;
56 VarPtr mul0_x_;
57 VarPtr mul1_x_;
58 VarPtr mul2_x_;
59 VarPtr mul3_x_;
60 VarPtr mul4_x_;
[all …]
Dlamb_next_right_rule.h44 VarPtr input0_;
45 VarPtr input1_;
46 VarPtr mul2_x_;
47 VarPtr mul3_x_;
48 VarPtr true_div1_recip_;
49 VarPtr add2_y_;
50 VarPtr add1_var_;
Dlamb_next_mv_with_decay_rule.h56 std::vector<VarPtr> input_vars_;
57 std::vector<VarPtr> constant_mul_input_vars_;
59 VarPtr constant_add2_y_;
60 VarPtr mul4_var_;
61 VarPtr real_div0_var_;
62 VarPtr real_div1_var_;
64 VarPtr add0_var_;
65 VarPtr add1_var_;
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Dadam_weight_decay_fusion.h45 VarPtr beta1_;
46 VarPtr one_sub_beta1_;
47 VarPtr beta2_;
48 VarPtr one_sub_beta2_;
49 VarPtr eps_;
50 VarPtr lr_;
51 VarPtr weight_decay_;
52 VarPtr param_;
53 VarPtr m_;
54 VarPtr v_;
[all …]
Dadam_fusion.h44 VarPtr beta1_;
45 VarPtr one_sub_beta1_;
46 VarPtr beta2_;
47 VarPtr one_sub_beta2_;
48 VarPtr eps_;
49 VarPtr lr_;
50 VarPtr param_;
51 VarPtr m_;
52 VarPtr v_;
53 VarPtr gradient_;
[all …]
Dbatch_norm_add_relu_grad_fusion.h44 VarPtr dy_;
45 VarPtr y_;
46 VarPtr x_;
47 VarPtr scale_;
48 VarPtr bias_;
49 VarPtr mean_;
50 VarPtr var_;
51 VarPtr save_mean_;
52 VarPtr save_var_;
53 VarPtr reserve_;
Dapply_momentum_weight_scale_fusion.h46 VarPtr monad_;
47 VarPtr weight_decay_;
48 VarPtr scale_;
49 VarPtr variable_;
50 VarPtr accumulation_;
51 VarPtr learning_rate_;
52 VarPtr cast_gradient_;
53 VarPtr momentum_;
54 VarPtr monad_state_;
Dapply_momentum_weight_fusion.h42 VarPtr monad_;
43 VarPtr weight_decay_;
44 VarPtr variable_;
45 VarPtr accumulation_;
46 VarPtr learning_rate_;
47 VarPtr gradient_;
48 VarPtr momentum_;
49 VarPtr monad_state_;
Dpost_batch_norm_add_relu_fusion.h41 VarPtr x_;
42 VarPtr scale_;
43 VarPtr bias_;
44 VarPtr mean_;
45 VarPtr var_;
46 VarPtr index_;
47 VarPtr z_;
Dbatch_norm_relu_grad_fusion.h41 VarPtr dy_;
42 VarPtr y_;
43 VarPtr x_;
44 VarPtr scale_;
45 VarPtr save_mean_;
46 VarPtr save_var_;
47 VarPtr reserve_;
Dbatch_norm_add_relu_fusion.h41 VarPtr x_;
42 VarPtr scale_;
43 VarPtr bias_;
44 VarPtr mean_;
45 VarPtr var_;
46 VarPtr index_;
47 VarPtr z_;
Dapply_momentum_scale_fusion.h43 VarPtr scale_;
44 VarPtr variable_;
45 VarPtr accumulation_;
46 VarPtr learning_rate_;
47 VarPtr gradient_;
48 VarPtr momentum_;
49 VarPtr monad_state_;
Dbatch_norm_relu_fusion.h39 VarPtr x_;
40 VarPtr scale_;
41 VarPtr bias_;
42 VarPtr mean_;
43 VarPtr var_;
44 VarPtr index_;
/third_party/mindspore/tests/ut/cpp/pre_activate/common/
Dpattern_engine_test.cc47 VarPtr v1 = std::make_shared<Var>(); in TEST_F()
48 VarPtr v2 = std::make_shared<Var>(); in TEST_F()
49 VarPtr v3 = std::make_shared<Var>("name"); in TEST_F()
72 VarPtr fv1 = std::make_shared<CondVar>(Pos); in TEST_F()
85 VarPtr sv1 = std::make_shared<SeqVar>(); in TEST_F()
105 VarPtr v1 = std::make_shared<Var>(); in TEST_F()
106 VarPtr v2 = std::make_shared<Var>(); in TEST_F()
107 VarPtr v3 = std::make_shared<Var>(); in TEST_F()
139 VarPtr v1 = std::make_shared<Var>(); in TEST_F()
140 VarPtr sv1 = std::make_shared<SeqVar>(); in TEST_F()
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/
Dpattern_engine.h94 VarNode(const VarPtr &value, const FuncGraphPtr &func_graph) : AnfNode(func_graph), var_(value) {} in VarNode()
98 const VarPtr var_;
134 explicit SeqVar(const VarPtr subvar) : subvar_(nullptr) { subvar_ = subvar; } in SeqVar()
150 VarPtr subvar_;
153 bool operator==(const VarPtr &lhs, const VarPtr &rhs);
155 inline bool operator!=(const VarPtr &lhs, const VarPtr &rhs) { return !(lhs == rhs); }
157 std::ostream &operator<<(std::ostream &os, const VarPtr &var);
159 using Equiv = std::map<VarPtr, BaseRef>;
161 using PrimitiveVarMap = std::unordered_map<PrimitivePtr, VarPtr>;
191 struct hash<mindspore::VarPtr> {
[all …]
Dpattern_engine.cc40 bool operator==(const VarPtr &lhs, const VarPtr &rhs) { in operator ==()
61 std::ostream &operator<<(std::ostream &os, const VarPtr &var) { in operator <<()
71 std::ostream &operator<<<VarPtr, BaseRef>(std::ostream &os, const Equiv &equiv) {
91 } else if (utils::isa<VarPtr>(x)) {
93 os << (utils::cast<VarPtr>(x));
120 } else if (utils::isa<VarPtr>(x)) { in GetVar()
130 if (utils::isa<VarPtr>(pattern)) { in MatchOnVar()
131 VarPtr var = utils::cast<VarPtr>(pattern); in MatchOnVar()
169 if (utils::isa<VarPtr>(value) && utils::cast<VarPtr>(value)->isa<SeqVar>()) { in GetSVarStartIndex()
263 if (utils::isa<VarPtr>(pattern_ref)) { in Match()
[all …]

123456