Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/embedding_cache/
Ddevice_sparse_embedding_operation.h70 …eCache(const DeviceAddress *embed_device_address, void *ids, void *embedding_cache, size_t ids_num, in LookupDeviceCache()
82 …bool EraseDeviceCache(void *ids, size_t ids_num, void *embedding_cache, const DeviceAddress *embed… in EraseDeviceCache()
Dembedding_cache_prefetch_actor.cc940 …ingCachePrefetchActor::PullEembeddingsFromRemote(int32_t param_key, const int *ids, size_t ids_num, in PullEembeddingsFromRemote()
993 …dingCachePrefetchActor::DoPushEmbeddingsToRemote(int32_t param_key, const int *ids, size_t ids_num, in DoPushEmbeddingsToRemote()
1027 …eddingCachePrefetchActor::PushEmbeddingsToRemote(int32_t param_key, const int *ids, size_t ids_num, in PushEmbeddingsToRemote()
1064 bool EmbeddingCachePrefetchActor::PartitionIds(const int *ids, size_t ids_num, in PartitionIds()
1097 bool EmbeddingCachePrefetchActor::PartitionIdsAndEmbeddings(const int *ids, size_t ids_num, const f… in PartitionIdsAndEmbeddings()
1167 int64_t ids_num = SizeToLong(keys_len / sizeof(int)); in SendToRemote() local
1216 const int *ids, size_t ids_num, const std::vector<std::vector<int>> &slice_ids_list, in RetrieveEmbeddings()