Home
last modified time | relevance | path

Searched defs:device_to_host_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.h91 std::unique_ptr<int[]> device_to_host_index; member
110 std::unique_ptr<int[]> device_to_host_index; member
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/embedding_cache/
Ddevice_embedding_operation.cc121 int *device_to_host_index = embedding_host_cache->device_to_host_index.get(); in ParseHostDataDeviceToHost() local
Ddevice_dense_embedding_operation.cc496 int *device_to_host_index = embedding_device_cache->device_to_host_index.get(); in ParseDeviceData() local