| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/optimizer/ |
| D | kernel_infer_resize_actor_insert.cc | 27 …lInferActorPtr CreateKernelInferActor(ActorSet *const actor_set, const KernelActor *kernel_actor) { in CreateKernelInferActor()
44 …esizeActorPtr CreateKernelResizeActor(ActorSet *const actor_set, const KernelActor *kernel_actor) { in CreateKernelResizeActor()
91 void AddDataArrows(const ActorSet *actor_set, const KernelActor *kernel_actor, KernelInferActor *ke… in AddDataArrows()
162 void AddControlArrows(KernelActor *kernel_actor, KernelInferActor *kernel_infer_actor, in AddControlArrows()
203 auto kernel_actor = dynamic_cast<const KernelActor *>(actor); in MatchPattern() local
215 auto kernel_actor = dynamic_cast<KernelActor *>(actor); in Process() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/ |
| D | super_kernel_actor.cc | 321 const auto &kernel_actor = kernel_actors_[i]; in UpdateMemoryTraceMangerStatus() local
375 void SuperKernelActor::SetTraceMemoryForKernel(const KernelActorPtr &kernel_actor) { in SetTraceMemoryForKernel()
445 const auto &kernel_actor = kernel_actors_[i]; in RunGraphKernelByKernel() local
831 KernelActorPtr kernel_actor = std::make_shared<KernelActor>( in BuildKernelActors() local
878 const auto &kernel_actor = kernel_actors_[i]; in BuildKernelActors() local
918 const auto &kernel_actor = kernel_actors_[i]; in ParseInputIndex() local
|
| D | kernel_async_launch_actor.cc | 33 …AsyncLaunchActor::LaunchKernel(OpContext<DeviceTensor> *const context, KernelActor *kernel_actor) { in LaunchKernel()
|
| D | kernel_async_infer_actor.cc | 33 …nelAsyncInferActor::InferShape(OpContext<DeviceTensor> *const context, KernelActor *kernel_actor) { in InferShape()
|
| D | kernel_async_resize_actor.cc | 33 …ncResizeActor::ResizeKernelMod(OpContext<DeviceTensor> *const context, KernelActor *kernel_actor) { in ResizeKernelMod()
|
| D | actor_dump.cc | 711 for (const auto &kernel_actor : actors) { in DumpKernelActors() local
771 auto kernel_actor = dynamic_cast<const KernelActor *>(actor.get()); in DumpNoInputKernelActors() local
831 const auto &kernel_actor = dynamic_cast<KernelActor *>(actor); in GetActorSubName() local
1058 const auto &kernel_actor = dynamic_cast<KernelActor *>(actor); in FetchOutputInfo() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/ |
| D | scheduler_helper.cc | 60 bool IsSkipLaunchShapeRelatedOp(KernelActor *kernel_actor) { in IsSkipLaunchShapeRelatedOp()
174 for (auto &kernel_actor : actor_set->kernel_actors_) { in CollectActors() local
311 auto kernel_actor = dynamic_cast<KernelActor *>(to_actor); in IsIgnoredInputAddress() local
849 auto kernel_actor = dynamic_cast<KernelActor *>(actor); in FetchKernelGraphByActor() local
921 auto kernel_actor = dynamic_cast<KernelActor *>(actor); in AddSomasInfo() local
968 auto kernel_actor = dynamic_cast<KernelActor *>(output_actor); in AddSomasInfoForGraphOutput() local
975 for (const auto &kernel_actor : kernel_actors) { in CheckKernelActorValid() local
1124 auto kernel_actor = dynamic_cast<KernelActor *>(actor.get()); in CheckActorValid() local
|
| D | inline_control_flow_scheduler.cc | 31 const auto &kernel_actor = dynamic_cast<KernelActor *>(actor.get()); in IsInlineKernelActor() local
45 std::string GetBranchNameByKernelActor(const KernelActor *const kernel_actor) { in GetBranchNameByKernelActor()
345 void InlineControlFlowScheduler::HandleConditionSwitchActor(const KernelActorPtr &kernel_actor) { in HandleConditionSwitchActor()
650 void InlineControlFlowScheduler::HandleConditionGatherActor(const KernelActorPtr &kernel_actor) { in HandleConditionGatherActor()
711 for (const auto &kernel_actor : actor_set->kernel_actors_) { in LinkControlArrowForNoInputOrOutputActor() local
750 for (const auto &kernel_actor : actor_set->kernel_actors_) { in Link() local
|
| D | rpc_node_scheduler.cc | 78 for (const auto &kernel_actor : kernel_actors) { in Build() local
|
| D | graph_scheduler.cc | 1369 const KernelActorPtr &kernel_actor, bool is_send_node) { in ProcessStreamSendRecvEventPair()
1420 KernelActorPtr kernel_actor = nullptr; in BuildKernelActor() local
1622 for (auto &kernel_actor : actor_set->kernel_actors_) { in BuildNoInputKernelActor() local
1882 const auto &kernel_actor = FetchActor(GetActorIdByKernel(kernel)); in LinkDataArrowInNonSinkMode() local
2789 for (auto &kernel_actor : actor_set->kernel_actors_) { in LinkControlArrowForLoopCountActor() local
2915 auto kernel_actor = dynamic_cast<KernelActor *>(auto_monad_actor); in LinkDeviceTensorStoreForAutoMonadActor() local
|
| D | memory_swap_node_scheduler.cc | 519 AbstractActor *kernel_actor = nullptr; in LinkControlArrowBySwapActor() local
|
| D | control_node_scheduler.cc | 1519 for (auto &kernel_actor : actor_set->kernel_actors_) { in LinkOutputControlArrowForActor() local
1585 const auto &kernel_actor = dynamic_cast<KernelActor *>(no_input_kernel_actor.get()); in LinkControlArrowForKernelActor() local
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/runtime/graph_scheduler/ |
| D | scheduler_helper_test.cc | 77 auto kernel_actor = in TEST_F() local
|