Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/include/backend/distributed/embedding_cache/
Dembedding_cache_utils.h93 std::unique_ptr<int[]> host_to_device_index; member
109 std::unique_ptr<int[]> host_to_device_index; member
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/embedding_cache/
Ddevice_embedding_operation.cc41 int *host_to_device_index = embedding_host_cache->host_to_device_index.get(); in ParseHostDataHostToDevice() local
Ddevice_dense_embedding_operation.cc498 int *host_to_device_index = embedding_device_cache->host_to_device_index.get(); in ParseDeviceData() local