Searched defs:embedding_cache (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/embedding_cache/ |
| D | device_sparse_embedding_operation.h | 70 …DeviceCache(const DeviceAddress *embed_device_address, void *ids, void *embedding_cache, size_t id… in LookupDeviceCache()
77 void *embedding_cache, const DeviceAddress *embed_device_address) { in UpdateDeviceCache()
82 …bool EraseDeviceCache(void *ids, size_t ids_num, void *embedding_cache, const DeviceAddress *embed… in EraseDeviceCache()
|
| D | device_dense_embedding_operation.cc | 285 bool DeviceDenseEmbeddingOperation::LookupDeviceCache(void *indices, void *embedding_cache, size_t … in LookupDeviceCache()
347 … size_t cache_size, size_t embedding_size, void *embedding_cache) { in UpdateDeviceCache()
|