Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/embedding_cache/
Ddevice_embedding_operation.cc120 int *device_to_host_ids = embedding_device_cache->device_to_host_ids.get(); in ParseHostDataDeviceToHost() local
Ddevice_dense_embedding_operation.cc497 int *device_to_host_ids = embedding_device_cache->device_to_host_ids.get(); in ParseDeviceData() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/include/backend/distributed/embedding_cache/
Dembedding_cache_utils.h92 std::unique_ptr<int[]> device_to_host_ids; member