Searched defs:batch_ids_num (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/embedding_cache/ |
| D | device_dense_embedding_operation.cc | 28 bool DeviceDenseEmbeddingOperation::AnalyseCache(int *batch_ids, const size_t batch_ids_num, size_t… in AnalyseCache()
410 …eDenseEmbeddingOperation::CheckCacheHitOrOutRange(const int *batch_ids, const size_t batch_ids_num, in CheckCacheHitOrOutRange()
445 …seEmbeddingOperation::CheckCacheHitOrOutRangeFunc(const int *batch_ids, const size_t batch_ids_num, in CheckCacheHitOrOutRangeFunc()
|
| D | device_embedding_operation.h | 74 virtual bool AnalyseCache(int *batch_ids, const size_t batch_ids_num, size_t data_step, in AnalyseCache()
|
| D | embedding_cache_prefetch_actor.cc | 179 …IdsToIndices(mindspore::HashMap<int, int> *unique_ids_to_indices, size_t batch_ids_num, int *batch… in TransformIdsToIndices()
1055 size_t batch_ids_num = (count != batch_size - 1) ? batch_num : batch_remainder; in PushEmbeddingsToRemote() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/distributed/embedding_cache/ |
| D | embedding_cache_utils.cc | 524 EmbeddingDeviceCache::EmbeddingDeviceCache(size_t batch_ids_num) { in EmbeddingDeviceCache()
531 EmbeddingHostCache::EmbeddingHostCache(size_t batch_ids_num) { in EmbeddingHostCache()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/include/backend/distributed/embedding_cache/ |
| D | embedding_cache_utils.h | 249 void set_batch_ids_num(size_t batch_ids_num) { batch_ids_num_ = batch_ids_num; } in set_batch_ids_num()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/ |
| D | embedding_cache_scheduler.cc | 303 …auto batch_ids_num = LongToSize(std::accumulate(shape.begin(), shape.end(), 1, std::multiplies<int… in ParseBatchIdsNum() local
|