Home
last modified time | relevance | path

Searched defs:EmbeddingCache (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/parameter_cache/
Dembedding_cache.h32 …EmbeddingCache(size_t vocab_size, size_t device_cache_size, size_t batch_elements, int rank_id, in… in EmbeddingCache() function
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/parameter_cache/
Dembedding_cache.h33 …EmbeddingCache(size_t vocab_size, size_t device_cache_size, size_t batch_elements, int rank_id, in… in EmbeddingCache() function