Home
last modified time | relevance | path

Searched defs:graph_compiler_info (Results 1 – 11 of 11) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/
Dgraph_scheduler.cc102 bool GetNeedSyncStream(const GraphCompilerInfo &graph_compiler_info) { in GetNeedSyncStream()
121 int64_t GetLoopCount(const GraphCompilerInfo &graph_compiler_info) { in GetLoopCount()
306 void ChangeGraphMode(const GraphCompilerInfo &graph_compiler_info) { in ChangeGraphMode()
566 ActorSet *GraphScheduler::Transform(const GraphCompilerInfo &graph_compiler_info) { in Transform()
993 ActorSetPtr GraphScheduler::Build(const GraphCompilerInfo &graph_compiler_info) { in Build()
1019 void GraphScheduler::CacheGraphOutputToActor(const GraphCompilerInfo &graph_compiler_info) { in CacheGraphOutputToActor()
1085 …hScheduler::UpdateDeviceAddressByRefInternalParameter(const GraphCompilerInfo &graph_compiler_info… in UpdateDeviceAddressByRefInternalParameter()
1152 void GraphScheduler::Link(ActorSet *actor_set, const GraphCompilerInfo &graph_compiler_info) { in Link()
1159 auto is_include_rpc = [](const GraphCompilerInfo &graph_compiler_info) { in Link()
1249 …aSourceActorPtr> GraphScheduler::BuildDataSourceActor(const GraphCompilerInfo &graph_compiler_info, in BuildDataSourceActor()
[all …]
Dany_type_graph_scheduler.cc34 …r<AnyTypeKernelActorPtr> AnyTypeGraphScheduler::Build(const GraphCompilerInfo &graph_compiler_info, in Build()
380 … const GraphCompilerInfo &graph_compiler_info, in FixDeviceTensorStoreKeyInActor()
487 auto graph_compiler_info = ConstructGraphCompilerInfo(model_graph, real_graph, device_context); in Transform() local
Dmem_swap_scheduler.cc345 …std::vector<MemSwapActorPtr>> MemSwapScheduler::Build(const GraphCompilerInfo &graph_compiler_info, in Build()
482 void MemSwapScheduler::Link(const GraphCompilerInfo &graph_compiler_info, ActorSet *actor_set) cons… in Link() argument
Dmemory_swap_node_scheduler.cc94 …ctor<MemSwapActorPtr>> MemorySwapNodeScheduler::Build(const GraphCompilerInfo &graph_compiler_info, in Build()
479 void MemorySwapNodeScheduler::Link(const GraphCompilerInfo &graph_compiler_info, ActorSet *actor_se… in Link() argument
Dcontrol_node_scheduler.cc123 ControlActorSetPtr ControlNodeScheduler::Build(const GraphCompilerInfo &graph_compiler_info, in Build()
166 const GraphCompilerInfo &graph_compiler_info, const HostTensorQueuePtr &host_queue, in BuildDataSourceActorForControlNode()
501 …d ControlNodeScheduler::BuildStackActorForControlNode(const GraphCompilerInfo &graph_compiler_info, in BuildStackActorForControlNode()
Dinline_control_flow_scheduler.cc738 …InlineControlFlowScheduler::Link(ActorSet *actor_set, const GraphCompilerInfo &graph_compiler_info, in Link()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/
Ddata_prepare_actor.h44 const AID *profiler_aid, const GraphCompilerInfo *graph_compiler_info, in DataPrepareActor()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/graph_compiler/
Dbackend_base.h104 …void RunGraphByCondition(const ActorInfo &actor_info, const GraphCompilerInfo &graph_compiler_info, in RunGraphByCondition()
Dbackend_base.cc144 …tor<std::vector<tensor::TensorPtr>> GetRunGraphInputs(const GraphCompilerInfo &graph_compiler_info, in GetRunGraphInputs()
554 auto graph_compiler_info = ConstructGraphCompilerInfo(root_graph); in CompileGraphs() local
1060 …dRTBackendBase::ContiguousArgs(const VectorRef &args, const GraphCompilerInfo &graph_compiler_info… in ContiguousArgs()
1085 void MindRTBackendBase::WaitMultiStream(const GraphCompilerInfo &graph_compiler_info) { in WaitMultiStream()
1118 const auto &graph_compiler_info = *(graph_iter->second); in RunGraph() local
Dbackend.cc648 …torSet *MindRTBackend::RealCompileGraphBeforeRunActor(const GraphCompilerInfo &graph_compiler_info, in RealCompileGraphBeforeRunActor()
707 …ackend::RunGraphByActors(const ActorInfo &actor_info, const GraphCompilerInfo &graph_compiler_info, in RunGraphByActors()
771 void MindRTBackend::RunGraphBySingleOp(const GraphCompilerInfo &graph_compiler_info, const VectorRe… in RunGraphBySingleOp()
870 …end::RunGraphByCondition(const ActorInfo &actor_info, const GraphCompilerInfo &graph_compiler_info, in RunGraphByCondition()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/pynative/
Dgraph_adapter.cc429 bool GraphAdapter::IsPynativeGeGraphSink(const GraphCompilerInfo &graph_compiler_info) { in IsPynativeGeGraphSink()