Home
last modified time | relevance | path

Searched refs:GetValueNode (Results 1 – 25 of 188) sorted by relevance

12345678

/third_party/mindspore/mindspore/lite/src/ops/
Dops_utils.cc27 auto prim = GetValueNode<std::shared_ptr<Primitive>>(node); in GetPrimitiveT()
49 auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Abs>>(node); in AbsPrimitiveCreator()
53 auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::AbsGrad>>(node); in AbsGradPrimitiveCreator()
57 auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Activation>>(node); in ActivationPrimitiveCreator()
61 auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::ActivationGrad>>(node); in ActivationGradPrimitiveCreator()
65 auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::Adam>>(node); in AdamPrimitiveCreator()
69 auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::AdderFusion>>(node); in AdderFusionPrimitiveCreator()
73 auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::AddFusion>>(node); in AddFusionPrimitiveCreator()
77 auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::AddGrad>>(node); in AddGradPrimitiveCreator()
81 auto ms_primc = GetValueNode<std::shared_ptr<mindspore::ops::AddN>>(node); in AddNPrimitiveCreator()
[all …]
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/
Dsymbol_resolver.cc48 auto ns = GetValueNode<parse::NameSpacePtr>(namespace_node); in operator ()()
49 auto sym = GetValueNode<parse::SymbolPtr>(symbol_node); in operator ()()
58 auto ns = GetValueNode<parse::NameSpacePtr>(ns_node.GetNode(node)); in operator ()()
59 auto str = GetValue<std::string>(GetValueNode(attr_node.GetNode(node))); in operator ()()
65 auto ns = GetValueNode<parse::NameSpacePtr>(ns_node.GetNode(node)); in operator ()()
66 auto sym = GetValueNode<parse::SymbolPtr>(sym_node.GetNode(node)); in operator ()()
Denv_item_eliminate.h74 auto key2 = GetValueNode<SymbolicKeyInstancePtr>(inputs[2]); in operator()
119 auto fg_inner = GetValueNode<FuncGraphPtr>(output_outer); in operator()
138 auto key2 = GetValueNode<SymbolicKeyInstancePtr>(inputs[2]); in operator()
166 (GetValueNode<EnvInstancePtr>(c1.GetNode(node)))->Len() == 0)); in operator()
241 auto c2 = GetValueNode<SymbolicKeyInstancePtr>(key2); in operator()
246 auto c1 = GetValueNode<SymbolicKeyInstancePtr>(inp1->input(2)); in operator()
267 auto symbolic_c1 = GetValueNode<SymbolicKeyInstancePtr>(inputs[2]); in operator()
325 auto key = GetValueNode<SymbolicKeyInstancePtr>(cnode->input(2)); in operator()
330 auto fg = GetValueNode<FuncGraphPtr>(inputs[0]); in operator()
378 auto key = GetValueNode<SymbolicKeyInstancePtr>(cnode->input(2)); in operator()
[all …]
Dcall_graph_tuple_transform.h51 auto fg = GetValueNode<FuncGraphPtr>(inputs[0]); in operator()
113 FuncGraphPtr fg = GetValueNode<FuncGraphPtr>(node); in TransformBranchNode()
124 FuncGraphPtr fg = GetValueNode<FuncGraphPtr>(partial_inputs[1]); in TransformBranchNode()
195 FuncGraphPtr fg = GetValueNode<FuncGraphPtr>(tuple_inputs[i]); in TransformLayerNode()
Dgradient_eliminate.cc31 auto prim = GetValueNode<PrimitivePtr>(vnode); in ExpandJPrimitive()
44 auto func_graph = GetValueNode<FuncGraphPtr>(value_node); in CheckIfEmbedJ()
89 auto func_graph = GetValueNode<FuncGraphPtr>(vnode); in ExpandJ()
Dminmax_grad.h39 auto prim = GetValueNode<PrimitivePtr>(cnode->input(0)); in IsOriginMaxMinGrad()
71 auto prim = GetValueNode<PrimitivePtr>(inputs[0]); in operator()
Dinline.h46 auto fg = GetValueNode<FuncGraphPtr>(node); in operator()
77 … (IsValueNode<FuncGraph>(inner) && GetValueNode<FuncGraphPtr>(inner)->parent() == nullptr)) { in operator()
114 auto fg = GetValueNode<FuncGraphPtr>(inputs[0]); in operator()
332 FuncGraphPtr call_fg = GetValueNode<FuncGraphPtr>(cinputs[0]); in GraphHasBranch()
343 FuncGraphPtr call_fg = GetValueNode<FuncGraphPtr>(cinputs[1]); in GraphHasBranch()
/third_party/mindspore/mindspore/lite/tools/optimizer/format/
Ddelete_redundant_transpose.cc36 auto sub_func_graph = GetValueNode<FuncGraphPtr>(cnode->input(1)); in DeleteNot4DTranspose()
45 sub_func_graph = GetValueNode<FuncGraphPtr>(cnode->input(kInputIndexTwo)); in DeleteNot4DTranspose()
97 auto sub_func_graph = GetValueNode<FuncGraphPtr>(cnode->input(1)); in TransTransFusion()
106 sub_func_graph = GetValueNode<FuncGraphPtr>(cnode->input(kInputIndexTwo)); in TransTransFusion()
146 auto prim = GetValueNode<PrimitivePtr>(cnode->input(0)); in UpdateNodeFormat()
162 auto post_prim = GetValueNode<PrimitivePtr>(post_cnode->input(0)); in UpdateNodeFormat()
Dto_format_base.cc49 auto trans_prim = GetValueNode<PrimitivePtr>(trans_cnode->input(0)); in GenNewInput()
70 auto prim = GetValueNode<PrimitivePtr>(cnode->input(0)); in ModifyCNode()
113 auto prim = GetValueNode<PrimitivePtr>(prim_node); in InsertPreTransNode()
195 auto prim = GetValueNode<PrimitivePtr>(post_cnode->input(0)); in DecideWhetherHandleGraphInput()
240 auto trans_prim = GetValueNode<PrimitivePtr>(trans_cnode->input(0)); in HandleGraphInput()
302 auto sub_func_graph = GetValueNode<FuncGraphPtr>(cnode->input(1)); in BasicProcess()
311 sub_func_graph = GetValueNode<FuncGraphPtr>(cnode->input(kInputIndexTwo)); in BasicProcess()
348 auto sub_func_graph = GetValueNode<FuncGraphPtr>(cnode->input(1)); in ConvWeightFormatTrans()
357 sub_func_graph = GetValueNode<FuncGraphPtr>(cnode->input(kInputIndexTwo)); in ConvWeightFormatTrans()
Dto_nhwc_format.cc24 auto prim = GetValueNode<PrimitivePtr>(prim_node); in GetTransNodeFormatType()
48 auto prim = GetValueNode<PrimitivePtr>(cnode->input(0)); in DecideConvWeightSrcAndDstFormat()
Dto_nchw_format.cc24 auto prim = GetValueNode<PrimitivePtr>(prim_node); in GetTransNodeFormatType()
48 auto prim = GetValueNode<PrimitivePtr>(cnode->input(0)); in DecideConvWeightSrcAndDstFormat()
/third_party/mindspore/mindspore/lite/tools/converter/parser/
Dunify_format.cc38 auto prim = GetValueNode<PrimitivePtr>(cnode->input(0)); in DecideMINDIRConvWeightSrcFormat()
58 auto prim = GetValueNode<PrimitivePtr>(cnode->input(0)); in DecideTFConvWeightSrcFormat()
81 auto prim = GetValueNode<PrimitivePtr>(cnode->input(0)); in DecideTFLITEConvWeightSrcFormat()
110 auto prim = GetValueNode<PrimitivePtr>(cnode->input(0)); in DecideONNXConvWeightSrcFormat()
138 auto prim = GetValueNode<PrimitivePtr>(prim_node); in GetTransNodeFormatType()
283 auto gather_prim = GetValueNode<PrimitivePtr>(gather_cnode->input(0)); in ConvertOnnxResizeForVariableShape()
293 auto prim = GetValueNode<PrimitivePtr>(cnode->input(0)); in ConvertOnnxResizeForVariableShape()
341 auto prim = GetValueNode<PrimitivePtr>(cnode->input(0)); in ProcessResizeAndFormat()
349 auto sub_func_graph = GetValueNode<FuncGraphPtr>(cnode->input(kNumIndex_1)); in ProcessResizeAndFormat()
355 sub_func_graph = GetValueNode<FuncGraphPtr>(cnode->input(kNumIndex_2)); in ProcessResizeAndFormat()
Dunused_node_remove_pass.cc45 if (utils::isa<ValueNode>(node) && GetValueNode<FuncGraphPtr>(node) != nullptr) { in ProcessGraph()
46 auto sub_graph = GetValueNode<FuncGraphPtr>(node); in ProcessGraph()
/third_party/mindspore/mindspore/lite/tools/optimizer/graph/
Dinfershape_pass.cc47 auto primitive = GetValueNode<PrimitivePtr>(real_cnode->input(0)); in GetCNodeCertainInputFormat()
89 auto primitive = GetValueNode<PrimitivePtr>(post_cnode->input(0)); in ModifySubGraphInputCNodeFormat()
141 auto sub_func_graph = GetValueNode<FuncGraphPtr>(cnode->input(1)); in JudgeAllOpsCanInfer()
148 sub_func_graph = GetValueNode<FuncGraphPtr>(cnode->input(kInputIndexTwo)); in JudgeAllOpsCanInfer()
159 auto prim = GetValueNode<PrimitivePtr>(cnode->input(0)); in JudgeAllOpsCanInfer()
182 auto sub_func_graph = GetValueNode<FuncGraphPtr>(cnode->input(1)); in InferProcess()
200 sub_func_graph = GetValueNode<FuncGraphPtr>(cnode->input(kInputIndexTwo)); in InferProcess()
257 auto out_prim = GetValueNode<PrimitivePtr>(out_cnode->input(0)); in SetSubGraphInput()
359 auto input_prim = GetValueNode<PrimitivePtr>(input_cnode->input(0)); in SetSubGraphAbstract()
377 auto prim = GetValueNode<PrimitivePtr>(cnode->input(0)); in SetSubGraphAbstract()
Ddecrease_transpose_algo.cc140 auto middle_node_prim = GetValueNode<PrimitivePtr>(middle_cnode->input(0)); in JudgeCanOptimizerForMultiOp()
280 auto prim = GetValueNode<PrimitivePtr>(prim_node); in InsertPreTransNode()
500 auto out_prim = GetValueNode<PrimitivePtr>(out_cnode->input(0)); in SetSubGraphInput()
607 auto input_prim = GetValueNode<PrimitivePtr>(input_cnode->input(0)); in SetSubGraphAbstract()
623 auto prim = GetValueNode<PrimitivePtr>(cnode->input(0)); in SetSubGraphAbstract()
635 auto primitive = GetValueNode<PrimitivePtr>(cnode->input(0)); in ModifyCNodeFormat()
664 auto sub_func_graph = GetValueNode<FuncGraphPtr>(cnode->input(1)); in DecreaseTransposeForSingleOp()
680 sub_func_graph = GetValueNode<FuncGraphPtr>(cnode->input(kInputIndexTwo)); in DecreaseTransposeForSingleOp()
703 auto prim = GetValueNode<PrimitivePtr>(cnode->input(0)); in DecreaseTransposeForSingleOp()
744 auto sub_func_graph = GetValueNode<FuncGraphPtr>(cnode->input(1)); in DecreaseTransposeForMultiOp()
[all …]
/third_party/mindspore/mindspore/lite/tools/converter/import/
Dprimitive_adjust.cc213 auto src_prim = GetValueNode<PrimitivePtr>(value_node); in MoveAttrMapCommon()
229 auto src_prim = GetValueNode<PrimitivePtr>(value_node); in MoveAttrMapActivation()
251 auto src_prim = GetValueNode<PrimitivePtr>(value_node); in MoveAttrMapActivationGrad()
273 auto src_prim = GetValueNode<PrimitivePtr>(value_node); in MoveAttrMapReduce()
296 auto src_prim = GetValueNode<PrimitivePtr>(value_node); in MoveAttrMapConv2D()
344 auto src_prim = GetValueNode<PrimitivePtr>(value_node); in MoveAttrPool()
380 auto src_prim = GetValueNode<PrimitivePtr>(value_node); in MoveAttrPoolGrad()
416 auto src_prim = GetValueNode<PrimitivePtr>(value_node); in MoveAttrMapAdder()
447 auto src_prim = GetValueNode<PrimitivePtr>(value_node); in MoveAttrMapLayerNorm()
467 auto src_prim = GetValueNode<PrimitivePtr>(value_node); in MoveAttrMapResize()
[all …]
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/pipeline_transformer/
Dpipeline_transformer.cc81 auto value = GetValueNode(cnode->input(2)); in SetMicroBatch()
127 auto sub_graph = GetValueNode<FuncGraphPtr>(cnode->input(0)); in LabelParameterStart()
209 auto graph = GetValueNode<FuncGraphPtr>(node); in Coloring()
263 auto prim = GetValueNode<PrimitivePtr>(cnode->input(0)); in IsPipelineCareNode()
293 auto output = GetValueNode<FuncGraphPtr>(cnode->input(0))->output(); in CreateOpInfo()
307 auto prim = GetValueNode<PrimitivePtr>(temp_node->input(0)); in CreateOpInfo()
318 input_value.push_back(GetValueNode(inputs[index])); in CreateOpInfo()
368 auto graph = GetValueNode<FuncGraphPtr>(temp_cnode->input(0)); in GetActualOpUsers()
424 graph = GetValueNode<FuncGraphPtr>(cnode->input(0)); in HandleSharedParameter()
468 graph = GetValueNode<FuncGraphPtr>(node->input(0)); in ParameterColoring()
[all …]
/third_party/mindspore/mindspore/lite/tools/converter/quantizer/
Dweight_quantizer.cc89 auto primitive = GetValueNode<PrimitivePtr>(cnode->input(0)); in DoConvQuantize()
149 auto primitive = GetValueNode<PrimitivePtr>(cnode->input(0)); in DoMulQuantize()
194 auto primitive = GetValueNode<PrimitivePtr>(cnode->input(0)); in DoLstmQuantize()
225 auto primitive = GetValueNode<PrimitivePtr>(cnode->input(0)); in DoGatherQuantize()
269 auto primitive = GetValueNode<PrimitivePtr>(cnode->input(0)); in DoOptimizerQuantize()
317 auto primitive = GetValueNode<PrimitivePtr>(cnode->input(0)); in DoMarkWeightQuantizeIfQuantized()
396 auto primitive = GetValueNode<std::shared_ptr<ops::PrimitiveC>>(cnode->input(0)); in MarkWeightQuantizationInNodes()
415 auto primitive = GetValueNode<std::shared_ptr<ops::PrimitiveC>>(cnode->input(0)); in DoQuantize()
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/
Dremove_value_node_dup.cc34 const auto &to_check_value = GetValueNode(node); in TryToDoReplace()
61 const auto &existed_value = GetValueNode(v); in TryToDoReplace()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/
Dnode_pass.cc43 auto switch_subgraph = GetValueNode<FuncGraphPtr>(partial_inputs.at(kPartialArgsIndex)); in AddOutputAndCallerToMap()
47 auto call_subgraph = GetValueNode<FuncGraphPtr>(inputs.at(kCallArgsIndex)); in AddOutputAndCallerToMap()
96 auto const_func_graph = GetValueNode<FuncGraphPtr>(new_node); in Run()
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/
Dcse.cc160 auto main_value = GetValueNode(main); in CheckReplace()
161 auto node_value = GetValueNode(node); in CheckReplace()
187 auto tensor1 = GetValueNode<tensor::TensorPtr>(inp1_j); in CheckReplace()
188 auto tensor2 = GetValueNode<tensor::TensorPtr>(inp2_j); in CheckReplace()
/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Dconv_activation_fusion.cc47 auto act_prim = GetValueNode<std::shared_ptr<mindspore::ops::Activation>>(act_node->input(0)); in Process()
65 auto prim = GetValueNode<PrimitivePtr>(conv_node->input(0)); in Process()
Daffine_activation_fusion.cc53 …auto activation_prim = GetValueNode<std::shared_ptr<ops::Activation>>(activation_node->input(kAnfP… in Process()
70 …auto affine_prim = GetValueNode<std::shared_ptr<ops::Affine>>(affine_node->input(kAnfPrimitiveInde… in Process()
Dconv_tuple_activation_fusion.cc53 auto act_prim = GetValueNode<std::shared_ptr<mindspore::ops::Activation>>(act_node->input(0)); in Process()
76 … auto primc = GetValueNode<std::shared_ptr<mindspore::ops::Conv2DFusion>>(conv_cnode->input(0)); in Process()
/third_party/mindspore/mindspore/ccsrc/vm/
Dvmimpl.cc68 return GetValueNode(node); in operator []()
102 if (GetValueNode(ct) == g) { in ComputeFvs()
251 (IsValueNode<FuncGraph>(i) && GetValueNode<FuncGraphPtr>(i)->parent() == graph)) { in SuccVm()
258 if (IsValueNode<FuncGraph>(node) && GetValueNode<FuncGraphPtr>(node)->parent() == graph) { in SuccVm()
259 auto fvs = utils::cast<SetRef>(vars_[GetValueNode<FuncGraphPtr>(node)]); in SuccVm()
400 auto g = GetValueNode<FuncGraphPtr>(node); in HandleNode()

12345678