Home
last modified time | relevance | path

Searched defs:root_graph (Results 1 – 10 of 10) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/trt_pass/
Dgraph_converter.cc126 …le<std::map<size_t, size_t>, CNodePtr> GraphConverter::BuildTrtNode(const FuncGraphPtr &root_graph, in BuildTrtNode()
/third_party/mindspore/mindspore/lite/tools/converter/parser/tf/
Dtf_model_parser.cc423 … std::unordered_map<std::string, AnfNodePtr> *anf_node_map, bool root_graph) { in ConvertParameter()
477 bool root_graph) { in ConvertGraphInputsAndConsts()
/third_party/mindspore/mindspore/ccsrc/vm/
Dbackend.cc371 auto root_graph = WrapPrimitives(func_graph); in CompileGraphs() local
946 …_ptr<GraphCompilerInfo> MindRTBackend::ConstructGraphCompilerInfo(const FuncGraphPtr &root_graph) { in ConstructGraphCompilerInfo()
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/
Dless_batch_normalization.cc324 auto root_graph = *(roots.begin()); in RemoveBatchNormalizetionNotUseParameters() local
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dgpu_session.cc387 auto root_graph = ConstructKernelGraph(func_graph, &all_graphs); in CompileGraphImpl() local
Dascend_session.cc539 auto root_graph = ConstructKernelGraph(func_graph, &all_graphs); in CompileGraphImpl() local
1789 void AscendSession::SelectKernel(NotNull<KernelGraphPtr> root_graph) { in SelectKernel()
Dsession_basic.cc457 void UpdateGraphAquireGilAttr(const NotNull<KernelGraphPtr> &root_graph) { in UpdateGraphAquireGilAttr()
2345 void SessionBasic::UpdateGraphDynamicShapeAttr(const NotNull<KernelGraphPtr> &root_graph) { in UpdateGraphDynamicShapeAttr()
Danf_runtime_algorithm.cc2274 void AnfRuntimeAlgorithm::InsertMakeTupleForOutput(const NotNull<KernelGraphPtr> &root_graph) { in InsertMakeTupleForOutput()
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/
Daction.cc294 FuncGraphPtr root_graph = *(manager->roots().begin()); in CheckRootInputShapeAndType() local
/third_party/mindspore/mindspore/ccsrc/runtime/framework/
Dcontrol_node_parser.cc723 … const std::vector<DeviceContext *> &device_contexts, const FuncGraphPtr &root_graph) { in Parse()