Home
last modified time | relevance | path

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

12

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/core/
Dgraph_kernel_splitter.cc203 auto sub_func_graph = std::make_shared<FuncGraph>(); in SplitGraph() local
278 CNodePtr CreateMainCNode(const FuncGraphPtr &main_func_graph, const FuncGraphPtr &sub_func_graph, in CreateMainCNode()
380 auto sub_func_graph = GetCNodeFuncGraph(cnode); in BindFuncGraph() local
393 auto sub_func_graph = GetCNodeFuncGraph(cnode); in RecoverParameter() local
503 auto sub_func_graph = GetCNodeFuncGraph(node); in UpdateMainNodesKernelInfo() local
544 auto sub_func_graph = GetCNodeFuncGraph(old_subgraph_cnode_); in GenParamMap() local
Dupdate_state_formatter.cc144 auto sub_func_graph = GetCNodeFuncGraph(node); in Run() local
231 …putForUpdateState::ProcessIndex(const FuncGraphPtr &func_graph, const FuncGraphPtr &sub_func_graph, in ProcessIndex()
Deliminate_redundant_output.cc149 bool CheckRepeatedOutput(const FuncGraphPtr &sub_func_graph) { in CheckRepeatedOutput()
Dtransform_op_optimizer.cc754 auto sub_func_graph = GetCNodeFuncGraph(node); in Run() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/format/
Ddelete_redundant_transpose.cc32 auto sub_func_graph = GetValueNode<FuncGraphPtr>(cnode->input(1)); in DeleteControlFlowTranspose() local
176 auto sub_func_graph = GetValueNode<FuncGraphPtr>(cnode->input(1)); in TransTransFusion() local
Dto_format_base.cc435 auto sub_func_graph = GetValueNode<FuncGraphPtr>(cnode->input(1)); in BasicProcess() local
488 auto sub_func_graph = GetValueNode<FuncGraphPtr>(cnode->input(1)); in ConvWeightFormatTrans() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/
Dconvert_custom_for_ge.cc81 FuncGraphPtr sub_func_graph = common::AnfAlgo::GetCNodeFuncGraphPtr(node); in SaveNodesInfo() local
Draise_reduction_precision.cc121 auto sub_func_graph = common::AnfAlgo::GetCNodeFuncGraphPtr(node); in Run() local
Daxis_normalizer.cc84 auto sub_func_graph = common::AnfAlgo::GetCNodeFuncGraphPtr(graph_kernel_node); in Process() local
Dtensor_inplace.cc257 auto sub_func_graph = common::AnfAlgo::GetCNodeFuncGraphPtr(node); in Run() local
Ddecrease_compute_precision.cc262 auto sub_func_graph = common::AnfAlgo::GetCNodeFuncGraphPtr(node); in Run() local
Dgraph_kernel_build.cc158 FuncGraphPtr sub_func_graph = common::AnfAlgo::GetCNodeFuncGraphPtr(node); in CollectNode() local
Dreorder_ops.cc364 auto sub_func_graph = common::AnfAlgo::GetCNodeFuncGraphPtr(node); in Run() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/
Dspecial_node_postprocess.cc62 auto sub_func_graph = GetValueNode<FuncGraphPtr>(cnode->input(1)); in Run() local
Dinfershape_pass.cc218 auto sub_func_graph = GetValueNode<FuncGraphPtr>(cnode->input(1)); in JudgeAllOpsCanInfer() local
247 auto sub_func_graph = GetValueNode<FuncGraphPtr>(cnode->input(1)); in InferProcessSubGraph() local
Dcore_infershape_pass.cc182 auto sub_func_graph = GetValueNode<FuncGraphPtr>(cnode->input(1)); in InferProcessSubGraph() local
Ddecrease_transpose_algo.cc749 auto sub_func_graph = GetValueNode<FuncGraphPtr>(cnode->input(1)); in DecreaseTransposeForSingleOp() local
827 auto sub_func_graph = GetValueNode<FuncGraphPtr>(cnode->input(1)); in DecreaseTransposeForMultiOp() local
Dredundant_op_remove_pass.cc515 auto sub_func_graph = GetValueNode<FuncGraphPtr>(node->cast<CNodePtr>()->input(1)); in RemoveRedundantOp() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/tf/
Dtf_model_parser.cc657 const FuncGraphPtr &sub_func_graph) { in ConvertSubgraphInputs()
702 const FuncGraphPtr &sub_func_graph) { in ConvertSubgraphOutputs()
750 void TFModelParser::UpdateMap(const CNodePtr &cnode, const FuncGraphPtr &sub_func_graph, in UpdateMap()
814 FuncGraphPtr sub_func_graph = std::make_shared<FuncGraph>(); in ConvertSubgraph() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/
Dexport_model.cc211 auto sub_func_graph = GetValueNode<FuncGraphPtr>(origin_input); in CloneFuncGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/
Dunify_format.cc399 auto sub_func_graph = GetValueNode<FuncGraphPtr>(cnode->input(kNumIndex_1)); in ProcessResizeAndFormat() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/tflite/
Dtflite_model_parser.cc638 …US TfliteModelParser::BuildSubFuncGraphMap(size_t subgraph_idx, const FuncGraphPtr &sub_func_graph, in BuildSubFuncGraphMap()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/irpass/
Dvmap_eliminate.cc684 auto sub_func_graph = GetValueNode<FuncGraphPtr>(node); in ExpandVmapValueNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/
Dparameter_manager.cc845 auto sub_func_graph = node_param_ptr->func_graph(); in RefParameterToActualParameter() local
Dstep_parallel.cc2235 auto sub_func_graph = node_param_ptr->func_graph(); in RefParameterToActualNode() local
3423 auto sub_func_graph = param_anf_node->func_graph(); in MoveMicroMirrorOutCallFunc() local

12