Searched defs:device_cache_size (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/parameter_cache/ |
| D | embedding_cache_manager.cc | 28 …heManager::Init(const std::string &cache_model_path, size_t vocab_size, size_t device_cache_size) { in Init()
53 size_t device_cache_size) { in Init()
|
| D | embedding_cache.h | 33 …EmbeddingCache(size_t vocab_size, size_t device_cache_size, size_t batch_elements, int rank_id, in… in EmbeddingCache()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/parameter_cache/ |
| D | embedding_cache_manager.cc | 28 …heManager::Init(const std::string &cache_model_path, size_t vocab_size, size_t device_cache_size) { in Init()
53 size_t device_cache_size) { in Init()
|
| D | embedding_cache.h | 32 …EmbeddingCache(size_t vocab_size, size_t device_cache_size, size_t batch_elements, int rank_id, in… in EmbeddingCache()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/ |
| D | lite_session.cc | 1014 size_t device_cache_size = 0; in CreateTensorRTDelegate() local
|