Searched defs:EmbeddingDeviceCache (Results 1 – 2 of 2) sorted by relevance
88 struct EmbeddingDeviceCache { struct91 std::unique_ptr<int[]> device_to_host_index;92 std::unique_ptr<int[]> device_to_host_ids;93 std::unique_ptr<int[]> host_to_device_index;94 std::unique_ptr<int[]> host_to_device_ids;
524 EmbeddingDeviceCache::EmbeddingDeviceCache(size_t batch_ids_num) { in EmbeddingDeviceCache() function in mindspore::distributed::EmbeddingDeviceCache