Home
last modified time | relevance | path

Searched defs:sub_func_graph (Results 1 – 18 of 18) sorted by relevance

/third_party/mindspore/mindspore/lite/tools/optimizer/format/
Ddelete_redundant_transpose.cc36 auto sub_func_graph = GetValueNode<FuncGraphPtr>(cnode->input(1)); in DeleteNot4DTranspose() local
97 auto sub_func_graph = GetValueNode<FuncGraphPtr>(cnode->input(1)); in TransTransFusion() local
Dto_format_base.cc302 auto sub_func_graph = GetValueNode<FuncGraphPtr>(cnode->input(1)); in BasicProcess() local
348 auto sub_func_graph = GetValueNode<FuncGraphPtr>(cnode->input(1)); in ConvWeightFormatTrans() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dgraph_kernel_splitter.cc343 auto sub_func_graph = std::make_shared<FuncGraph>(); in SplitGraph() local
410 CNodePtr CreateMainCNode(const FuncGraphPtr &main_func_graph, const FuncGraphPtr &sub_func_graph, in CreateMainCNode()
531 auto sub_func_graph = AnfAlgo::GetCNodeFuncGraphPtr(cnode); in BindFuncGraph() local
544 auto sub_func_graph = AnfAlgo::GetCNodeFuncGraphPtr(cnode); in RecoverParameter() local
633 auto sub_func_graph = AnfAlgo::GetCNodeFuncGraphPtr(cnode); in UpdateSubGraphInfo() local
674 auto sub_func_graph = AnfAlgo::GetCNodeFuncGraphPtr(old_subgraph_cnode_); in GenParamMap() local
Daxis_normalizer.cc86 auto sub_func_graph = AnfAlgo::GetCNodeFuncGraphPtr(node); in Run() local
Dupdate_state_formatter.cc143 auto sub_func_graph = AnfAlgo::GetCNodeFuncGraphPtr(node); in Run() local
222 …putForUpdateState::ProcessIndex(const FuncGraphPtr &func_graph, const FuncGraphPtr &sub_func_graph, in ProcessIndex()
Draise_reduction_precision.cc116 auto sub_func_graph = AnfAlgo::GetCNodeFuncGraphPtr(node); in Run() local
Ddecrease_compute_precision.cc254 auto sub_func_graph = AnfAlgo::GetCNodeFuncGraphPtr(node); in Run() local
Deliminate_redundant_output.cc150 bool CheckRepeatedOutput(const FuncGraphPtr &sub_func_graph) { in CheckRepeatedOutput()
Dreorder_ops.cc360 auto sub_func_graph = AnfAlgo::GetCNodeFuncGraphPtr(node); in Run() local
Dtransform_op_optimizer.cc432 auto sub_func_graph = AnfAlgo::GetCNodeFuncGraphPtr(node); in Run() local
/third_party/mindspore/mindspore/lite/tools/optimizer/graph/
Dinfershape_pass.cc141 auto sub_func_graph = GetValueNode<FuncGraphPtr>(cnode->input(1)); in JudgeAllOpsCanInfer() local
182 auto sub_func_graph = GetValueNode<FuncGraphPtr>(cnode->input(1)); in InferProcess() local
Ddecrease_transpose_algo.cc664 auto sub_func_graph = GetValueNode<FuncGraphPtr>(cnode->input(1)); in DecreaseTransposeForSingleOp() local
744 auto sub_func_graph = GetValueNode<FuncGraphPtr>(cnode->input(1)); in DecreaseTransposeForMultiOp() local
799 auto sub_func_graph = GetValueNode<FuncGraphPtr>(cnode->input(1)); in DoFixFormat() local
Dredundant_op_remove_pass.cc393 auto sub_func_graph = GetValueNode<FuncGraphPtr>(node->cast<CNodePtr>()->input(1)); in Run() local
/third_party/mindspore/mindspore/lite/tools/converter/parser/tf/
Dtf_model_parser.cc605 const FuncGraphPtr &sub_func_graph) { in ConvertSubgraphInputs()
646 const FuncGraphPtr &sub_func_graph) { in ConvertSubgraphOutputs()
692 void TFModelParser::UpdateMap(const CNodePtr &cnode, const FuncGraphPtr &sub_func_graph, in UpdateMap()
735 FuncGraphPtr sub_func_graph = std::make_shared<FuncGraph>(); in ConvertSubgraph() local
/third_party/mindspore/mindspore/lite/tools/converter/
Dexport_model.cc166 auto sub_func_graph = GetValueNode<FuncGraphPtr>(origin_input); in CloneFuncGraph() local
/third_party/mindspore/mindspore/lite/tools/converter/parser/
Dunify_format.cc349 auto sub_func_graph = GetValueNode<FuncGraphPtr>(cnode->input(kNumIndex_1)); in ProcessResizeAndFormat() local
/third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/
Dtflite_model_parser.cc549 …US TfliteModelParser::BuildSubFuncGraphMap(size_t subgraph_idx, const FuncGraphPtr &sub_func_graph, in BuildSubFuncGraphMap()
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dsession_basic.cc1079 auto sub_func_graph = AnfAlgo::GetValueNodeFuncGraph(anf); in CreateValueNodeKernelGraph() local