Home
last modified time | relevance | path

Searched defs:main_graph (Results 1 – 20 of 20) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pass/
Dbegin_end_overlap_inline.cc66 auto main_graph = expanding_node->func_graph(); in InlineExpandFuncGraph() local
82 auto main_graph = expanding_node->func_graph(); in InlineExpandPartialFuncGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/
Dtsa_atomic_add_to_first_tensor.cc88const KernelGraphPtr &main_graph, const InplaceAssignerInfo &atomic_add_info, const AnfNodePtr &no… in GetOrCreateNewTsaFirstNode() argument
207 void TsaAtomicAddToFirstTensor::ProcessTsa(const KernelGraphPtr &main_graph, const AnfNodePtr &anf_… in ProcessTsa()
Dinplace_assign_builder.cc122const FuncGraphPtr &main_graph, TypeId dst_type) { in CreateCleanCompositeNode()
264 const FuncGraphPtr &main_graph, const AnfNodePtr &composite_node, in ProcessOriginCNodeUser()
Dfloatstatus_addn_fusion.cc62 void FloatStatusAddNFusion::ProcessFloatStatusAddN(const FuncGraphPtr &main_graph, const CNodePtr &… in ProcessFloatStatusAddN()
Dadd_stitch_atomic_clean_gpu.cc57 void StitchAtomicCleanInserter::AddDepend(const FuncGraphPtr &main_graph, const AnfNodePtr &clean_n… in AddDepend()
Dadd_atomic_clean.cc236 void AtomicCleanInserter::InsertAtomicClean(const FuncGraphPtr &main_graph, const AnfNodePtr &anf_n… in InsertAtomicClean()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/
Danf_transform_for_ge.cc155 STATUS AnfTransformForGe::Transform(const FuncGraphPtr &main_graph, const std::shared_ptr<Converter… in Transform()
Danf_transform.cc847 STATUS AnfTransform::Transform(const FuncGraphPtr &main_graph, const std::shared_ptr<ConverterPara>… in Transform()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/core/
Dgraph_kernel_expander.cc28 auto main_graph = node->func_graph(); in CreateExpandedNode() local
Dgraph_builder.cc394 …odePtr ReplaceNodesWithGraphKernelNode(const AnfNodePtrList &nodes, const FuncGraphPtr &main_graph, in ReplaceNodesWithGraphKernelNode()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/parser/
Dunify_format.cc209 bool UnifyFormatToNHWC::BasicProcess(const api::FuncGraphPtr &func_graph, bool main_graph) { in BasicProcess()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/adapter/
Dexpander.cc365 auto main_graph = expanding_node->func_graph(); in InlineExpandFuncGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/format/
Dto_format_base.cc420 bool ToFormatBase::BasicProcess(const FuncGraphPtr &func_graph, bool main_graph) { in BasicProcess()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/
Dsub_graph_split.cc727 LiteGraph::SubGraph *main_graph = model_->graph_.sub_graphs_.front(); in InitMainGraphDevice() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/auto_parallel/
Dstage_compute.cc812 FuncGraphPtr main_graph; in ChangeStageNumber() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/hardware/
Dge_kernel_executor.cc254 auto main_graph = kernel_cnode->func_graph(); in InlineSubGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/train/optimizer/fusion/
Dgru_fusion_pass.cc363 auto &main_graph = graph_->subGraph.front(); in UpdateMetaGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/cache_embedding/
Dcache_embedding.cc414 void CreateControlDepend(const FuncGraphPtr &main_graph, const AnfNodePtr &prior_node, const AnfNod… in CreateControlDepend()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/tf/
Dtf_model_parser.cc866 auto main_graph = ConvertGraph(res_graph_); in ControlFlowNodePostProcess() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pipeline_transformer/
Dpipeline_interleave.cc1297 FuncGraphPtr main_graph; in IsolatedNodeAttach() local