Home
last modified time | relevance | path

Searched defs:device_tensor_store_keys (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/
Dabstract_actor.h131 const std::vector<std::pair<size_t, AnfNodePtr>> &device_tensor_store_keys() const { in device_tensor_store_keys() function
134 …ice_tensor_store_keys(const std::vector<std::pair<size_t, AnfNodePtr>> &device_tensor_store_keys) { in set_device_tensor_store_keys()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/
Dany_type_graph_scheduler.cc394 std::vector<std::pair<size_t, AnfNodePtr>> device_tensor_store_keys; in FixDeviceTensorStoreKeyInActor() local
Dscheduler_helper.cc295 const auto &device_tensor_store_keys = to_actor->device_tensor_store_keys_; in AddMonadDeviceTensorStore() local