Home
last modified time | relevance | path

Searched defs:data_arrow (Results 1 – 17 of 17) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/optimizer/
Dkernel_infer_resize_actor_insert.cc61 AbstractActor *FetchRealFromActor(const DataArrow *data_arrow, const ActorSet *actor_set, const CNo… in FetchRealFromActor()
103 const DataArrow *data_arrow = input_pair.second; in AddDataArrows() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/
Dcopy_actor.cc39 for (auto &data_arrow : output_data_arrows_) { in Init() local
Dactor_dump.cc73 for (const auto &data_arrow : batch_output_data_arrow.second) { in DumpBaseOutputInfo() local
92 auto data_arrow = actor->output_data_arrows()[i]; in DumpBaseOutputInfo() local
870 …romIndexInHostQueueDataSourceActor(AbstractActor *input_actor, const DataArrow *const data_arrow) { in GetFromIndexInHostQueueDataSourceActor()
895 const DataArrow *const data_arrow) { in GetFromIndexInSuperKernelActor()
940 const ActorInfoMap &actor_info, const DataArrow *const data_arrow) { in FetchInputActor()
Dabstract_actor.cc149 for (auto &data_arrow : output_data_arrows_) { in InitOutputData() local
Ddata_source_actor.cc59 …rceActor::UpdateOutputData(OpData<DeviceTensor> *const output_data, const DataArrowPtr &data_arrow, in UpdateOutputData()
Dabstract_actor.h178 …tual void UpdateOutputData(OpData<DeviceTensor> *const output_data, const DataArrowPtr &data_arrow, in UpdateOutputData()
Dany_type_kernel_actor.cc740 …nelActor::UpdateOutputData(OpData<DeviceTensor> *const output_data, const DataArrowPtr &data_arrow, in UpdateOutputData()
Dsuper_kernel_actor.cc113 auto &data_arrow = output_data_arrows_[i]; in Init() local
Dkernel_actor.cc103 for (auto &data_arrow : output_data_arrows_) { in Init() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/
Dinline_control_flow_scheduler.cc172 DataArrow *data_arrow, in GetBranchNameByConditionGatherActor()
228 const auto &data_arrow = condition_switch_actor->output_data_arrows()[i]; in InitOutputDataBranchInfoForConditionSwitchActor() local
555 const auto &data_arrow = pair.second; in InitInputDataBranchInfoForConditionGatherActor() local
778 [&ref_count, &output_pair](const auto &data_arrow) { in Link()
Dscheduler_helper.cc388 …auto data_arrow = std::make_shared<DataArrow>(from_output_index, to_actor->GetAID(), to_input_inde… in AddDataArrow() local
544 auto data_arrow = std::make_shared<DataArrow>(from_index, to_actor->GetAID(), to_index); in AddDataArrowForExitActor() local
612const DataArrowPtr &data_arrow, size_t data_arrow_index) { in ConvertDataArrowToControlArrow()
692 for (const auto &data_arrow : actor->output_data_arrows()) { in FuseDataArrowsToBatchDataArrow() local
698 for (auto &data_arrow : actor->output_data_arrows()) { in FuseDataArrowsToBatchDataArrow() local
Dany_type_graph_scheduler.cc251 for (auto &data_arrow : data_arrows) { in TransArrowInOutputActorToAnyTypeKernelActor() local
446 … auto data_arrow = std::make_shared<DataArrow>(from_index, fusion_actor->GetAID(), pair.first); in FixDeviceTensorStoreKeyInActor() local
464 auto data_arrow = std::make_shared<DataArrow>(from_index, actor->GetAID(), pair.first); in FixDeviceTensorStoreKeyInActor() local
Dcontrol_node_scheduler.cc858 const auto &data_arrow = input_pair.second; in GetLazyInlineFuncGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/
Dparallel_lite_actor.h70 …void AddOutputDataArrows(const DataArrowPtr &data_arrow) { output_data_arrows_.push_back(data_arro… in AddOutputDataArrows()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/control_flow/
Dcondition_switch_actor.cc48 const auto &data_arrow = output_data_arrows_[i]; in Init() local
Dcontrol_actor.cc41 for (auto &data_arrow : output_data_arrows_) { in Init() local
391 …rolActor::UpdateOutputData(OpData<DeviceTensor> *const output_data, const DataArrowPtr &data_arrow, in UpdateOutputData()
Dexit_actor.cc32 for (auto &data_arrow : output_branch_data_arrows) { in Init() local