Home
last modified time | relevance | path

Searched refs:NewValueNode (Results 1 – 25 of 280) sorted by relevance

12345678910>>...12

/third_party/mindspore/tests/ut/cpp/optimizer/ad/
Dad_test.cc95 FuncGraphPtr dg = Kprim(NewValueNode(prim::kPrimScalarAdd), resourcePtr); in TEST_F()
100 FuncGraphPtr dg = Kprim(NewValueNode(prim::kPrimScalarMul), resourcePtr); in TEST_F()
105 FuncGraphPtr dg = Kprim(NewValueNode(prim::kPrimScalarSub), resourcePtr); in TEST_F()
110 FuncGraphPtr dg = Kprim(NewValueNode(prim::kPrimScalarDiv), resourcePtr); in TEST_F()
115 FuncGraphPtr dg = Kprim(NewValueNode(prim::kPrimScalarPow), resourcePtr); in TEST_F()
120 FuncGraphPtr dg = Kprim(NewValueNode(prim::kPrimScalarExp), resourcePtr); in TEST_F()
125 FuncGraphPtr dg = Kprim(NewValueNode(prim::kPrimScalarUadd), resourcePtr); in TEST_F()
130 FuncGraphPtr dg = Kprim(NewValueNode(prim::kPrimScalarUsub), resourcePtr); in TEST_F()
135 FuncGraphPtr dg = Kprim(NewValueNode(prim::kPrimScalarGt), resourcePtr); in TEST_F()
140 FuncGraphPtr dg = Kprim(NewValueNode(prim::kPrimScalarLt), resourcePtr); in TEST_F()
[all …]
Dkpynative_test.cc52 …auto c_node = g->NewCNode({NewValueNode(prim::GetPythonOps("tensor_mul", "mindspore.ops.functional… in BuildPrimalFuncGraph()
70 …auto a_node = g->NewCNode({NewValueNode(prim::GetPythonOps("tensor_mul", "mindspore.ops.functional… in BuildStopGradient()
72 auto b_node = g->NewCNode({NewValueNode(prim::kPrimStopGradient), a_node}); in BuildStopGradient()
74 …auto c_node = g->NewCNode({NewValueNode(prim::GetPythonOps("tensor_mul", "mindspore.ops.functional… in BuildStopGradient()
77 …g->NewCNode({NewValueNode(prim::GetPythonOps("tensor_mul", "mindspore.ops.functional")), a_node, c… in BuildStopGradient()
/third_party/mindspore/mindspore/ccsrc/frontend/operator/composite/
Dcomposite.cc103 inputs.push_back(NewValueNode(fn_leaf_)); in FullMake()
139 auto fn_rec = NewValueNode(std::make_shared<HyperMap>(*this)); in FullMake()
143 inputs.push_back(NewValueNode(prim::kPrimMakeList)); in FullMake()
156 … {NewValueNode(prim::kPrimListGetItem), item.first, NewValueNode(SizeToLong(pos))}); in FullMake()
197 auto fn_rec = NewValueNode(std::make_shared<HyperMap>(*this)); in FullMake()
201 inputs.push_back(NewValueNode(prim::kPrimMakeTuple)); in FullMake()
214 … {NewValueNode(prim::kPrimTupleGetItem), item.first, NewValueNode(SizeToLong(pos))}); in FullMake()
236 inputs.push_back(NewValueNode(prim::kPrimMakeRecord)); in FullMake()
237 inputs.push_back(NewValueNode(type)); in FullMake()
241 auto fn_rec = NewValueNode(std::make_shared<HyperMap>(*this)); in FullMake()
[all …]
Dunpack_call.cc67 …ret_graph->NewCNode({NewValueNode(prim::kPrimTupleGetItem), para_tuple, NewValueNode(SizeToLong(i)… in GenerateFuncGraph()
74 …ret_graph->NewCNode({NewValueNode(prim::kPrimListGetItem), para_list, NewValueNode(SizeToLong(i))}… in GenerateFuncGraph()
83 … {NewValueNode(prim::kPrimDictGetItem), para_dict, NewValueNode(item.first)}); in GenerateFuncGraph()
85 … {NewValueNode(prim::kPrimMakeKeywordArg), NewValueNode(item.first), dict_get_item}); in GenerateFuncGraph()
Dmap.cc44 inputs.emplace_back(NewValueNode(fn_leaf_)); in FullMakeLeaf()
97 inputs.push_back(NewValueNode(prim::kPrimMakeList)); in FullMakeList()
102 auto fn = NewValueNode(ptrGraph); in FullMakeList()
114 … {NewValueNode(prim::kPrimListGetItem), item.first, NewValueNode(SizeToLong(pos))}); in FullMakeList()
156 inputs.push_back(NewValueNode(prim::kPrimMakeTuple)); in FullMakeTuple()
161 auto fn = NewValueNode(ptrGraph); in FullMakeTuple()
173 … {NewValueNode(prim::kPrimTupleGetItem), item.first, NewValueNode(SizeToLong(pos))}); in FullMakeTuple()
195 inputs.push_back(NewValueNode(prim::kPrimMakeRecord)); in FullMakeClass()
196 inputs.push_back(NewValueNode(type)); in FullMakeClass()
201 auto fn = NewValueNode(ptrGraph); in FullMakeClass()
[all …]
Dzip_operation.cc67 make_tuple_nodes.push_back(NewValueNode(prim::kPrimMakeTuple)); in GenerateFuncGraph()
70 make_tuple_zip_nodes.push_back(NewValueNode(prim::kPrimMakeTuple)); in GenerateFuncGraph()
74 … std::vector<AnfNodePtr> tuple_get_item_nodes{NewValueNode(op), ret_graph->parameters()[arg_idx], in GenerateFuncGraph()
75 NewValueNode(SizeToLong(idx))}; in GenerateFuncGraph()
/third_party/mindspore/tests/ut/cpp/session/
Danf_runtime_algorithm_test.cc45 ValueNodePtr value_node = NewValueNode(prim::kPrimAdd); in TEST_F()
64 std::vector<AnfNodePtr> add_inputs{NewValueNode(prim::kPrimAdd)}; in TEST_F()
66 std::vector<AnfNodePtr> make_tuple_inputs{NewValueNode(prim::kPrimMakeTuple), add, add_second}; in TEST_F()
82 std::vector<AnfNodePtr> tuple_get_item_inputs{NewValueNode(prim::kPrimTupleGetItem), make_tuple, in TEST_F()
83 NewValueNode(static_cast<int64_t>(1))}; in TEST_F()
90 std::vector<AnfNodePtr> depend_inputs{NewValueNode(prim::kPrimDepend), add, add_second}; in TEST_F()
102 std::vector<AnfNodePtr> inputs{NewValueNode(add_primitive)}; in TEST_F()
112 std::vector<AnfNodePtr> inputs{NewValueNode(prim::kPrimAdd)}; 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()
[all …]
Dkernel_graph_test.cc37 TEST_F(KernelGraphTest, NewValueNode) { in TEST_F() argument
39 auto add_value = NewValueNode(MakeValue(static_cast<int64_t>(0))); in TEST_F()
51 auto new_value = kernel_graph->NewValueNode(add_value); in TEST_F()
99 auto add_value = NewValueNode(prim::kPrimAdd); in TEST_F()
152 std::vector<AnfNodePtr> add_inputs = {NewValueNode(prim::kPrimAdd), x_parameter, y_parameter}; in TEST_F()
161 std::vector<AnfNodePtr> mul_inputs = {NewValueNode(prim::kPrimMul), add, z_parameter}; in TEST_F()
166 std::vector<AnfNodePtr> make_tuple_inputs = {NewValueNode(prim::kPrimMakeTuple), mul}; in TEST_F()
/third_party/mindspore/tests/ut/cpp/ir/
Dmanager_test.cc258 inputs.push_back(NewValueNode(prim)); in MakeFuncGraph()
263 inputs.push_back(NewValueNode(prim::kPrimReturn)); in MakeFuncGraph()
283 inputs.push_back(NewValueNode(fg)); in MakeNestedGraph()
284 inputs.push_back(NewValueNode(prim::kPrimReturn)); in MakeNestedGraph()
290 inputs.push_back(NewValueNode(prim::kPrimReturn)); in MakeNestedGraph()
317 inputs.push_back(NewValueNode(prim::kPrimScalarAdd)); in MakeNestedGraph2()
322 inputs.push_back(NewValueNode(prim::kPrimReturn)); in MakeNestedGraph2()
329 inputs.push_back(NewValueNode(graph_bar)); in MakeNestedGraph2()
334 inputs.push_back(NewValueNode(prim::kPrimReturn)); in MakeNestedGraph2()
363 inputs.push_back(NewValueNode(prim_scalar_add)); in TEST_F()
[all …]
/third_party/mindspore/tests/ut/cpp/pipeline/static_analysis/
Dspecialize_test.cc63 inputs.push_back(NewValueNode(prim_return)); in SetUp()
75 inputs.push_back(NewValueNode(graph_g_)); in SetUp()
79 inputs.push_back(NewValueNode(prim_return)); in SetUp()
99 inputs.push_back(NewValueNode(std::make_shared<Primitive>(prim::kScalarAdd))); in SetUp()
104 inputs.push_back(NewValueNode(std::make_shared<Primitive>("Return"))); in SetUp()
111 inputs.push_back(NewValueNode(graph_beta_)); in SetUp()
116 inputs.push_back(NewValueNode(prim_return)); in SetUp()
168 inputs.push_back(NewValueNode(prim_scalar_add)); in GenerateFromTypes()
174 inputs.push_back(NewValueNode(prim_return)); in GenerateFromTypes()
197 inputs.push_back(NewValueNode(meta_scalar_add)); in SetUp()
[all …]
Dstatic_analysis_test.cc102 inputs.push_back(NewValueNode(prim_scalar_add)); in GenerateFromTypes()
108 inputs.push_back(NewValueNode(prim_return)); in GenerateFromTypes()
139 inputs.push_back(NewValueNode(prim)); in MakeFuncGraph()
144 inputs.push_back(NewValueNode(prim::kPrimReturn)); in MakeFuncGraph()
196 inputs.push_back(NewValueNode(prim_return)); in SetUp()
208 inputs.push_back(NewValueNode(graph_g_)); in SetUp()
212 inputs.push_back(NewValueNode(prim_return)); in SetUp()
232 inputs.push_back(NewValueNode(prim::kPrimScalarAdd)); in SetUp()
237 inputs.push_back(NewValueNode(prim::kPrimReturn)); in SetUp()
244 inputs.push_back(NewValueNode(graph_beta_)); in SetUp()
[all …]
/third_party/mindspore/tests/ut/cpp/optimizer/
Dclean_test.cc53 AnfNodePtr valuenode_next = NewValueNode(std::string("ms_next")); in SetUp()
71 …AnfNodePtr cnode_67 = me_graph->NewCNode({NewValueNode(prim::kPrimPartial), valuenode_next, parame… in SetUp()
74 …AnfNodePtr cnode_66 = me_graph->NewCNode({NewValueNode(prim::kPrimScalarAdd), cnode_57, cnode_67}); in SetUp()
77 AnfNodePtr valuenode_return = NewValueNode(prim::kPrimReturn); in SetUp()
136 auto cons_make_record = NewValueNode(prim::kPrimMakeRecord); in TEST_F()
146 auto cons_class = NewValueNode(abs_ptr->BuildValue()); in TEST_F()
152 auto cons_return = NewValueNode(prim::kPrimReturn); in TEST_F()
171 auto cons_partial = NewValueNode(prim::kPrimPartial); in TEST_F()
175 auto cons_make_record = NewValueNode(prim::kPrimMakeRecord); in TEST_F()
181 auto cons_class = NewValueNode(abs_ptr->BuildValue()); in TEST_F()
[all …]
/third_party/mindspore/mindspore/lite/tools/optimizer/graph/
Dadd_tensor_array.cc86 auto make_tuple_vnode = NewValueNode(make_tuple_prim_ptr); in SetGraphOutput()
101 auto return_value_node = NewValueNode(return_prim_ptr); in SetGraphOutput()
172 auto tensor_array_vnode = NewValueNode(tensor_array); in Process()
174 auto num_tensors_vnode = NewValueNode(kDefaultNumTensors); in Process()
184 auto tensor_array_read_vnode = NewValueNode(tensor_array_read); in Process()
186 auto read_index_vnode = NewValueNode(kDefaultIndex); in Process()
188 auto read_flow_in_vnode = NewValueNode(kFlowInPlaceHolder); in Process()
200 auto tensor_array_write_vnode = NewValueNode(tensor_array_write); in Process()
202 auto write_index_vnode = NewValueNode(kDefaultIndex); in Process()
204 auto write_flow_in_vnode = NewValueNode(kFlowInPlaceHolder); in Process()
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/
Dbranch_culling.cc36 std::vector<AnfNodePtr> switch_nodes{NewValueNode(switch_node), data, cond}; in GenerateSwitchNode()
38 std::vector<AnfNodePtr> tuple_getitem_nodes{NewValueNode(prim::kPrimTupleGetItem), switch_apply, in GenerateSwitchNode()
39 NewValueNode(MakeValue(switch_idx))}; in GenerateSwitchNode()
238 auto ctrl_data = NewValueNode(const_data); in SquareOp()
241 std::vector<AnfNodePtr> square_nodes{NewValueNode(PrimSquare), ctrl_node}; in SquareOp()
251 auto oppsite_ctrl_data = NewValueNode(const_data); in MergeNode()
256 merge_nodes.push_back(NewValueNode(PrimMerge)); in MergeNode()
257 …std::vector<AnfNodePtr> make_tuple_nodes{NewValueNode(prim::kPrimMakeTuple), square_op, opposite_c… in MergeNode()
281 AnfNodePtrList inputs = {NewValueNode(prim::kPrimDepend), square_op, output_node}; in GenerateSwitchDependNode()
333 new_make_tuple_nodes.push_back(NewValueNode(prim::kPrimMakeTuple)); in GenerateReplNodeForDependMakeTuple()
[all …]
Dincorporate_getitem.h65 auto idx_node = NewValueNode(idx); in operator()
67 … auto output_node = new_fg->NewCNode({NewValueNode(prim::kPrimTupleGetItem), output, idx_node}); in operator()
106 new_fg_outer->set_output(NewValueNode(new_fg)); in operator()
117 auto idx_node = NewValueNode(idx); in operator()
119 … auto output_node = new_fg->NewCNode({NewValueNode(prim::kPrimTupleGetItem), output, idx_node}); in operator()
184 txn.SetEdge(cnode_and_index.tp_cnode, kInputIndex, NewValueNode(i)); in UpdateUserNodeIndex()
233 AnfNodePtrList shrunk_inputs{NewValueNode({prim::kPrimMakeTuple})}; in ShrinkUnsedOutput()
259 shrunk_output = NewValueNode(std::make_shared<ValueTuple>(shrunk_inputs)); in ShrinkUnsedOutput()
428 manager->SetEdge(fg_call_cnode_, 0, NewValueNode(iter->second)); in ShrinkFuncGraphOutput()
435 …node->func_graph()->NewCNode({NewValueNode(prim::kPrimTupleGetItem), fg_call_cnode_, NewValueNode(… in ShrinkFuncGraphOutput()
[all …]
Denv_item_eliminate.h82 …new_fg->set_output(new_fg->NewCNode({NewValueNode(prim::kPrimEnvGetItem), env, NewValueNode(key), … in operator()
121 new_fg_outer->set_output(NewValueNode(new_fg)); in operator()
145 …new_fg->set_output(new_fg->NewCNode({NewValueNode(prim::kPrimEnvGetItem), env, NewValueNode(key), … in operator()
200 auto xcz = fg->NewCNode({NewValueNode(prim::kPrimEnvGetItem), x, c, z}); in operator()
201 auto ycz = fg->NewCNode({NewValueNode(prim::kPrimEnvGetItem), y, c, z}); in operator()
203 return fg->NewCNode({NewValueNode(PrimHyperAdd_), xcz, ycz}); in operator()
273 … return node->func_graph()->NewCNode({NewValueNode(prim::kPrimEnvGetItem), env, key2, default_v}); in operator()
340 args.push_back(NewValueNode(new_fg)); in operator()
400 …auto new_sw = fg->NewCNode({NewValueNode(prim::kPrimSwitch), x, NewValueNode(new_g1), NewValueNode… in operator()
471 layers.push_back(NewValueNode(fg_transform)); in operator()
[all …]
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/
Dclean.cc136 auto idx_c = NewValueNode(count); in ConvertGetAttrToTupleGetItem()
140 return node->func_graph()->NewCNode({NewValueNode(prim::kPrimTupleGetItem), data, idx_c}); in ConvertGetAttrToTupleGetItem()
177 auto idx_c = NewValueNode(count); in ConvertDictGetItemToTupleGetItem()
180 return node->func_graph()->NewCNode({NewValueNode(prim::kPrimTupleGetItem), data, idx_c}); in ConvertDictGetItemToTupleGetItem()
221 …auto tuple_new_item = node->func_graph()->NewCNode({NewValueNode(prim::kPrimMakeTuple), item_value… in ConvertDictSetItemToTupleSetItem()
222 return node->func_graph()->NewCNode({NewValueNode(tuple_add_op), data, tuple_new_item}); in ConvertDictSetItemToTupleSetItem()
224 auto idx_c = NewValueNode(count); in ConvertDictSetItemToTupleSetItem()
227 …return node->func_graph()->NewCNode({NewValueNode(prim::kPrimTupleSetItem), data, idx_c, item_valu… in ConvertDictSetItemToTupleSetItem()
235 inputs.emplace_back(NewValueNode(prim::kPrimMakeTuple)); in ConvertMakeRecordToMakeTuple()
256 return NewValueNode(prim::kPrimMakeTuple); in ErasePartialNode()
[all …]
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ad/
Ddfunctor.cc125 embed_node = tape_->NewCNode({NewValueNode(prim::kPrimEmbed), fv_node}); in BackPropagateFv()
126 default_val_node = tape_->NewCNode({NewValueNode(prim::GetPythonOps("zeros_like")), fv_node}); in BackPropagateFv()
131 …auto dfv = tape_->NewCNode({NewValueNode(prim::kPrimEnvGetItem), din, embed_node, default_val_node… in BackPropagateFv()
191 …k_graph_->NewCNode({NewValueNode(prim::kPrimTupleGetItem), k_app, NewValueNode(static_cast<int64_t… in BackPropagate()
205 …tape_->NewCNode({NewValueNode(prim::kPrimTupleGetItem), bprop_app, NewValueNode(static_cast<int64_… in BackPropagate()
210 …auto din = tape_->NewCNode({NewValueNode(prim::kPrimTupleGetItem), bprop_app, NewValueNode(SizeToL… in BackPropagate()
307 …k_graph_->NewCNode({NewValueNode(prim::kPrimTupleGetItem), k_app, NewValueNode(static_cast<int64_t… in MapMorphism()
381 auto node = tape_->NewCNode({NewValueNode(prim::kPrimEmbed), fv_adjoint->second->k()}); in AttachFvDoutToTape()
385 NewValueNode(prim::kPrimEnvSetItem), in AttachFvDoutToTape()
408 auto node = tape_->NewCNode({NewValueNode(prim::kPrimEmbed), fv_adjoint.second->k()}); in AttachIndirectFvDoutToTape()
[all …]
Dkpynative.cc120 auto zeros_like_node = tape->NewCNode({NewValueNode(zeros_like_fg), node}); in BuildZerosLikeNode()
129 auto zeros_like_value = tape->NewCNode({NewValueNode(zeros_like_fg), NewValueNode(out)}); in BuildZerosLikeValue()
163 auto ones_like_value = tape->NewCNode({NewValueNode(ones_like_fg), NewValueNode(out)}); in BuildOnesLikeValue()
172 outputs.push_back(NewValueNode(prim::kPrimMakeTuple)); in BuildFakeBProp()
176 auto fake_input_sens = func_graph->NewCNode({NewValueNode(fake_bprop), NewValueNode(true)}); in BuildFakeBProp()
223 dout_ = tape_->NewCNode({NewValueNode(add_fg), dout_, dout_factor}); in AccumulateDout()
693 …din = tape_->NewCNode({NewValueNode(prim::kPrimTupleGetItem), bprop_app, NewValueNode(SizeToLong(i… in BackPropagate()
697 …din = tape_->NewCNode({NewValueNode(prim::kPrimTupleGetItem), bprop_app, NewValueNode(SizeToLong(i… in BackPropagate()
720 return NewValueNode(cnode_adjoint->op_args()[input_index - 1]); in BuildKNodeForCNodeInput()
751 node_list.push_back(NewValueNode(bprop_fg)); in BackPropagateOneCNodeWithBPropFuncGraph()
[all …]
Dkprim.cc191 std::vector<AnfNodePtr> new_inputs{NewValueNode(python_ops_value)}; in GetPythonOps()
369 auto make_tuple = NewValueNode(prim::kPrimMakeTuple); in AppendMonadOutput()
379 auto u = NewValueNode(kUMonad); in AdjustForAutoMonad()
385 auto io = NewValueNode(kIOMonad); in AdjustForAutoMonad()
485 extra_node = NewValueNode(kUMonad); in BuildOutput()
487 extra_node = NewValueNode(kIOMonad); in BuildOutput()
505 args.push_back(NewValueNode(prim::kPrimMakeTuple)); in BuildOutput()
506 args.push_back(NewValueNode(newenv)); in BuildOutput()
517 auto tuple_env = NewCNode({NewValueNode(prim::kPrimMakeTuple), NewValueNode(newenv)}, bprop_fg); in BuildOutput()
518 auto tuple_add_ops = NewValueNode(prim::GetPythonOps(python_ops, model_name)); in BuildOutput()
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/
Ddynamic_rnn_grad_fission_v2.cc54 NewValueNode(std::make_shared<Primitive>(kBasicLSTMCellCStateGradV2OpName))}; in CreateTLoopNode()
68 …std::vector<AnfNodePtr> matmul_inputs = {NewValueNode(std::make_shared<Primitive>(prim::kPrimMatMu… in CreateTLoopNode()
76 …std::vector<AnfNodePtr> splitv_input = {NewValueNode(std::make_shared<Primitive>(prim::kPrimSplitV… in CreateTLoopNode()
106 …std::vector<AnfNodePtr> lstm_split_input = {NewValueNode(std::make_shared<Primitive>(prim::kPrimSp… in CreateLSTMSPlitV()
182 lstm_x_concat_input[0] = NewValueNode(std::make_shared<Primitive>(prim::kPrimConcat->name())); in AddLSTMInputGradNode()
184 lstm_gage_concat_input[0] = NewValueNode(std::make_shared<Primitive>(prim::kPrimConcat->name())); in AddLSTMInputGradNode()
190 NewValueNode(std::make_shared<Primitive>(kBasicLSTMCellCStateGradV2OpName))}; in AddLSTMInputGradNode()
192 …std::vector<AnfNodePtr> reshape_inputs = {NewValueNode(std::make_shared<Primitive>(prim::kPrimResh… in AddLSTMInputGradNode()
227 …std::vector<AnfNodePtr> matmul_inputs = {NewValueNode(std::make_shared<Primitive>(prim::kPrimMatMu… in AddLSTMInputGradNode()
236 …std::vector<AnfNodePtr> splitv_input = {NewValueNode(std::make_shared<Primitive>(prim::kPrimSplitV… in AddLSTMInputGradNode()
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/
Dsparse_softmax_cross_entropy_with_logits_unify_mindir.cc87 …one_hot_inputs = {NewValueNode(one_hot_primitive), sparse_softmax_node->input(kIndex2), value_on_n… in CreateOneHot()
90 auto depth_node = NewValueNode(depth); in CreateOneHot()
96 …one_hot_inputs = {NewValueNode(one_hot_primitive), sparse_softmax_node->input(kIndex2), depth_node… in CreateOneHot()
117 …std::vector<AnfNodePtr> inputs = {NewValueNode(std::make_shared<Primitive>(kSoftmaxCrossEntropyWit… in CreateSoftmaxCrossEntropyWithLogits()
180 inputs = {NewValueNode(reduce_primitive), softmax_output_node}; in CreateReduceMean()
183 inputs = {NewValueNode(reduce_primitive), softmax_output_node, axis_node}; in CreateReduceMean()
203 auto axis_node = NewValueNode(axis); in CreateExpandDims()
215 …std::vector<AnfNodePtr> expand_dims_inputs = {NewValueNode(expand_dims_primitive), real_div_node, … in CreateExpandDims()
238 std::vector<AnfNodePtr> expand_dims_inputs = {NewValueNode(expand_dims_primitive), real_div_node}; in CreateExpandDimsPynative()
278 tile_inputs = {NewValueNode(tile_primitive), mul_node->input(2)}; in CreateTile()
[all …]
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/parse/
Dfunction_block.cc254 ValueNodePtr module_node = NewValueNode(name_space); in MakeResolve()
255 ValueNodePtr symbol_node = NewValueNode(resolve_symbol); in MakeResolve()
256 …auto node = func_graph_->NewCNodeInOrder({NewValueNode(prim::kPrimResolve), module_node, symbol_no… in MakeResolve()
264 auto script_node = NewValueNode(script); in MakeInterpret()
266 {NewValueNode(prim::kPrimPyInterpret), script_node, global_dict_node, local_dict_node}); in MakeInterpret()
422 input_nodes.emplace_back(NewValueNode(target_block->func_graph())); in Jump()
442 …func_graph_->NewCNodeInOrder({NewValueNode(prim::kPrimSwitch), condNode, NewValueNode(true_block->… in ConditionalJump()
443 NewValueNode(false_block->func_graph())}); in ConditionalJump()
453 ValueNodePtr assign_op = NewValueNode(prim::GetPythonOps(primitive_name, module_name, true)); in SetStateAssign()
506 states.emplace_back(NewValueNode(prim::kPrimMakeTuple)); in AttachIsolatedNodesBeforeReturn()
[all …]
/third_party/mindspore/mindspore/core/ir/
Dfunc_graph_extends.cc61 std::vector<AnfNodePtr> params({NewValueNode(prim::kPrimReturn), value}); in set_output()
94 var_param_tuple_nodes.push_back(NewValueNode(prim::kPrimMakeTuple)); in GenerateVarParams()
124 std::vector<AnfNodePtr> kwarg_keys_tuple_nodes = {NewValueNode(prim::kPrimMakeTuple)}; in GenerateKwParams()
125 std::vector<AnfNodePtr> kwarg_values_tuple_nodes = {NewValueNode(prim::kPrimMakeTuple)}; in GenerateKwParams()
152 kwarg_keys_tuple_nodes.push_back(NewValueNode(kw_param_name)); in GenerateKwParams()
154 …specialized_graph->NewCNode({NewValueNode(prim::kPrimExtractKeywordArg), NewValueNode(kw_param_nam… in GenerateKwParams()
166 {NewValueNode(prim::kPrimExtractKeywordArg), NewValueNode(kw_param_name), param_node}); in GenerateKwParams()
187 …specialized_graph->NewCNode({NewValueNode(prim::kPrimMakeDict), make_tuple_keys, make_tuple_values… in GenerateKwargReplNode()
/third_party/mindspore/tests/ut/cpp/utils/
Dsymbolic_test.cc31 …auto sk1 = std::make_shared<SymbolicKeyInstance>(NewValueNode(static_cast<int64_t>(1)), abstract::… in TEST_F()
32 …auto sk1b = std::make_shared<SymbolicKeyInstance>(NewValueNode(static_cast<int64_t>(1)), abstract:… in TEST_F()
36 …auto sk2 = std::make_shared<SymbolicKeyInstance>(NewValueNode(static_cast<int64_t>(2)), abstract::… in TEST_F()

12345678910>>...12