Home
last modified time | relevance | path

Searched defs:sub_graph (Results 1 – 25 of 28) sorted by relevance

12

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dadd_stitch_atomic_clean_gpu.cc33 CNodePtr StitchAtomicCleanInsertter::CreateInplaceAssignNode(const FuncGraphPtr &sub_graph, in CreateInplaceAssignNode()
47 auto sub_graph = AnfAlgo::GetCNodeFuncGraphPtr(composite_node); in ProcessOriginCNode() local
95 auto sub_graph = AnfAlgo::GetCNodeFuncGraphPtr(node); in FindInnerCNodeUsers() local
112 auto sub_graph = AnfAlgo::GetCNodeFuncGraphPtr(node); in IsStitchWithAtomic() local
Dtsa_atomic_add_to_first_tensor.cc64 auto sub_graph = AnfAlgo::GetCNodeFuncGraphPtr(cnode); in FindTsaFirstRealInputInGraph() local
187 auto sub_graph = AnfAlgo::GetCNodeFuncGraphPtr(composite_node); in ProcessOriginCNode() local
Dshape_ops_splitter.cc121 auto sub_graph = AnfAlgo::GetCNodeFuncGraphPtr(anf_node); in Run() local
Dadd_atomic_clean.cc107 auto sub_graph = AnfAlgo::GetCNodeFuncGraphPtr(node); in FindCandidate() local
271 void AtomicCleanInsertter::CreateInplaceAssignNodeAndCorrectReturn(const FuncGraphPtr &sub_graph, in CreateInplaceAssignNodeAndCorrectReturn()
338 auto sub_graph = AnfAlgo::GetCNodeFuncGraphPtr(composite_node); in ProcessOriginCNode() local
Ddecrease_transfer_precision.cc70 auto sub_graph = AnfAlgo::GetCNodeFuncGraphPtr(node); in IsCandidateNode() local
Darithmetic_simplify.cc641 auto sub_graph = AnfAlgo::GetCNodeFuncGraphPtr(node); in Run() local
/third_party/mindspore/mindspore/lite/tools/optimizer/graph/
Dinfershape_pass.cc74 int ModifySubGraphInputCNodeFormat(const FuncGraphPtr &sub_graph, const ParameterPtr &certain_input, in ModifySubGraphInputCNodeFormat()
234 STATUS InferShapePass::SetSubGraphInput(const CNodePtr &cnode, const FuncGraphPtr &sub_graph) { in SetSubGraphInput()
300 STATUS InferShapePass::SetSubGraphOutput(const FuncGraphPtr &sub_graph) { in SetSubGraphOutput()
332 STATUS InferShapePass::SetSubGraphAbstract(const CNodePtr &cnode, const FuncGraphPtr &sub_graph) { in SetSubGraphAbstract()
385 auto &sub_graph = iter->first; in ResetSubGraphInput() local
Ddecrease_transpose_algo.cc476 int DecreaseTransposeAlgo::SetSubGraphInput(const CNodePtr &cnode, const FuncGraphPtr &sub_graph) { in SetSubGraphInput()
531 auto &sub_graph = iter->first; in ResetSubGraphInput() local
549 int DecreaseTransposeAlgo::SetSubGraphOutput(const FuncGraphPtr &sub_graph) { in SetSubGraphOutput()
580 int DecreaseTransposeAlgo::SetSubGraphAbstract(const CNodePtr &cnode, const FuncGraphPtr &sub_graph… in SetSubGraphAbstract()
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/opencl/
Dfill_tests.cc84 …auto *sub_graph = new (std::nothrow) kernel::OpenCLSubGraph(kernels, kernels, kernels, sub_inner_k… in TEST_F() local
154 …auto *sub_graph = new (std::nothrow) kernel::OpenCLSubGraph(kernels, kernels, kernels, sub_inner_k… in TEST_F() local
Dcast_tests.cc117 …auto *sub_graph = new (std::nothrow) kernel::OpenCLSubGraph(kernels, kernels, kernels, sub_inner_k… in TEST_F() local
227 …auto *sub_graph = new (std::nothrow) kernel::OpenCLSubGraph(kernels, kernels, kernels, sub_inner_k… in TEST_F() local
Dcommon.cc144 auto sub_graph = new (std::nothrow) OpenCLSubGraph(kernels, kernels, kernels, sub_inner_kernel); in TestMain() local
309 auto sub_graph = new (std::nothrow) OpenCLSubGraph(kernels, kernels, kernels, sub_inner_kernel); in TestMain() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/trt_pass/
Dgraph_converter.cc127 … const FuncGraphPtr &sub_graph, in BuildTrtNode()
197 FuncGraphPtr sub_graph; in ReplaceSubgraphWithTrtNode() local
263 const Subgraph &sub_graph = graph_partition->CreateNewGraph(segment.second); in Run() local
/third_party/mindspore/mindspore/lite/tools/converter/parser/
Dunused_node_remove_pass.cc46 auto sub_graph = GetValueNode<FuncGraphPtr>(node); in ProcessGraph() local
/third_party/mindspore/mindspore/lite/src/
Dlite_model.cc142 auto sub_graph = this->graph_.sub_graphs_[i]; in Destroy() local
147 int LiteModel::ConvertSubGraph(const schema::SubGraph &sub_graph) { in ConvertSubGraph()
Dlite_session.cc630 auto sub_graph = reinterpret_cast<kernel::SubGraphKernel *>(kernel); in PrepareKernels() local
940 auto sub_graph = reinterpret_cast<kernel::OpenCLSubGraph *>(kernel); in ReSizeKernels() local
944 auto sub_graph = reinterpret_cast<kernel::SubGraphKernel *>(kernel); in ReSizeKernels() local
Dlite_model.h237 auto sub_graph = sub_graphs->template GetAs<schema::SubGraph>(i); in GenerateModel() local
Dsub_graph_split.cc79 for (const auto &sub_graph : subgraphs) { in CheckIsParallelSubGraph() local
/third_party/mindspore/mindspore/ccsrc/debug/
Danf_ir_dump.cc385 void DumpShape(const AnfNodePtr &nd, const FuncGraphPtr &sub_graph, const std::shared_ptr<SubGraphI… in DumpShape()
405 void DumpCNode(const CNodePtr &nd, const FuncGraphPtr &sub_graph, OrderedMap<AnfNodePtr, int32_t> *… in DumpCNode()
491 FuncGraphPtr sub_graph = nd->func_graph(); in DumpIRInSubgraph() local
Ddraw.cc69 auto sub_graph = nd->func_graph(); in DrawNodes() local
/third_party/mindspore/mindspore/lite/tools/common/
Dfunc_graph_subgraph.cc477 const FuncGraphPtr &sub_graph, std::vector<AnfNodePtr> *partial_inputs, in CreateParameterForPartialSubGraph()
545 …const FuncGraphPtr &sub_graph, const std::map<AnfNodePtr, AnfNodePtr> &partial_inputs_and_subgraph… in CreateCNodeForPartialSubGraph()
/third_party/mindspore/mindspore/lite/src/train/
Dtrain_export.cc513 auto sub_graph = std::make_unique<schema::SubGraphT>(); in ExportInit() local
Dtrain_session.cc621 auto sub_graph = reinterpret_cast<kernel::SubGraphKernel *>(ori_kernel); in CompileTrainKernels() local
/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Dtflite_lstm_cell_fusion.cc362 auto sub_graph = GetValueNode<FuncGraphPtr>(anf_sub_graph); in CheckSubGraph() local
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dascend_auto_monad.cc543 auto &sub_graph = callee.graph; in DoSearchRecursiveCall() local
1610 for (auto &sub_graph : sub_graphs) { in GenerateExecuteOrder() local
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/pipeline_transformer/
Dpipeline_transformer.cc127 auto sub_graph = GetValueNode<FuncGraphPtr>(cnode->input(0)); in LabelParameterStart() local

12