Home
last modified time | relevance | path

Searched defs:device_tensor_store_key (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/framework/actor/
Doutput_actor.cc65 for (const auto &device_tensor_store_key : device_tensor_store_keys_) { in CollectLoopCount() local
Dgather_actor.cc151 for (const auto &device_tensor_store_key : device_tensor_store_keys_) { in FetchInputDeviceTensor() local
Dkernel_actor.cc323 for (auto &device_tensor_store_key : device_tensor_store_keys_) { in FetchInputDeviceTensor() local
Dswitch_actor.cc360 for (const auto &device_tensor_store_key : device_tensor_store_keys_) { in FetchInputDeviceTensor() local
/third_party/mindspore/mindspore/ccsrc/runtime/framework/
Dgraph_scheduler.cc949 auto device_tensor_store_key = FetchFrontNodeByBackendNode(from_kernel, graph); in LinkDataArrowForDeviceTensorStore() local
1596 for (auto &device_tensor_store_key : kernel_actor->device_tensor_store_keys_) { in LinkDeviceTensorStoreForAutoMonadActor() local
2434 for (const auto &device_tensor_store_key : actor->device_tensor_store_keys_) { in DumpAbstractActor() local