| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/cache_embedding/ |
| D | ps_embedding_cache_inserter.h | 34 …PsEmbeddingCacheInserter(const FuncGraphPtr &root_graph, int64_t rank_id, const std::string &node_… in PsEmbeddingCacheInserter()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/ |
| D | opt.h | 102 SimpleRewriter(const FuncGraphPtr &root_graph, const FuncGraphManagerPtr &manager) in SimpleRewriter()
|
| D | fallback_rewriter.cc | 131 BaseRewriter(const FuncGraphPtr &root_graph, const FuncGraphManagerPtr &manager) in BaseRewriter()
243 BeforeOptARewriter(const FuncGraphPtr &root_graph, const FuncGraphManagerPtr &manager) in BeforeOptARewriter()
864 AfterOptARewriter(const FuncGraphPtr &root_graph, const FuncGraphManagerPtr &manager, in AfterOptARewriter()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/trt_pass/ |
| D | graph_converter.cc | 126 …le<std::map<size_t, size_t>, CNodePtr> GraphConverter::BuildTrtNode(const FuncGraphPtr &root_graph, in BuildTrtNode()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/graph_compiler/ |
| D | backend_base.cc | 499 auto root_graph = WrapPrimitives(func_graph); in CompileGraphs() local
791 auto root_graph = func_graph; in CompileSubGraph() local
1026 const FuncGraphPtr &root_graph) { in ConstructOutputs()
1505 …<GraphCompilerInfo> MindRTBackendBase::ConstructGraphCompilerInfo(const FuncGraphPtr &root_graph) { in ConstructGraphCompilerInfo()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/ |
| D | graph_compiler.cc | 575 KernelGraphPtr root_graph = session_->ConstructKernelGraph(func_graph, all_graphs, device_target); in ConstructKernelGraphForGraphRunMode() local
642 KernelGraphPtr root_graph; in CompileWholeGraphForGraphRunMode() local
|
| D | control_node_scheduler.cc | 1497 const auto &root_graph = parser->root_func_graph_; in LinkControlArrowForLoopCountActor() local
2007 const auto &root_graph = parser->root_func_graph_; in LinkDataArrowForOutputActor() local
2037 const auto &root_graph = graph_compiler_info.control_node_parser_->root_func_graph_; in LinkArrowForRootGraphEntranceActor() local
|
| D | control_node_parser.cc | 992 … const std::vector<DeviceContext *> &device_contexts, const FuncGraphPtr &root_graph, in Parse()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/dynamic_shape/ |
| D | dynamic_shape.cc | 388 FuncGraphPtr root_graph = roots.back(); in InDynamicGraph() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/common/debug/ |
| D | mindir_exporter.cc | 230 ModelProtoPtr IrExporter::GetDumpProto(const FuncGraphPtr &root_graph, const std::vector<FuncGraphP… in GetDumpProto()
257 bool IrExportBuilder::BuildModel(const FuncGraphPtr &root_graph, const std::vector<FuncGraphPtr> &c… in BuildModel()
1790 bool DumpBinaryProto(const FuncGraphPtr &root_graph, const std::vector<FuncGraphPtr> &child_graphs, in DumpBinaryProto()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/hal/hardware/ |
| D | gpu_session.cc | 389 auto root_graph = ConstructKernelGraph(func_graph, &all_graphs, DeviceType::kGPU); in CompileGraphImpl() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/tf/ |
| D | tf_model_parser.cc | 470 … std::unordered_map<std::string, AnfNodePtr> *anf_node_map, bool root_graph) { in ConvertParameter()
523 bool root_graph) { in ConvertGraphInputsAndConsts()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | less_batch_normalization.cc | 332 auto root_graph = *(roots.begin()); in RemoveBatchNormalizetionNotUseParameters() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/session/ |
| D | anf_runtime_algorithm.cc | 1582 void AnfRuntimeAlgorithm::InsertMakeTupleForOutput(const NotNull<KernelGraphPtr> &root_graph) { in InsertMakeTupleForOutput()
|
| D | kernel_graph_mgr.cc | 721 bool DumpKernelGraphJson(const KernelGraphPtr &root_graph, const std::set<KernelGraphPtr> &child_gr… in DumpKernelGraphJson()
|