Searched defs:data_step (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/embedding_cache/ |
| D | device_dense_embedding_operation.cc | 28 …DenseEmbeddingOperation::AnalyseCache(int *batch_ids, const size_t batch_ids_num, size_t data_step, in AnalyseCache() 411 … int *hash_index, bool *out_range, size_t data_step) { in CheckCacheHitOrOutRange() 447 … size_t *hash_hit_count, size_t data_step) { in CheckCacheHitOrOutRangeFunc() 471 … bool *need_swap_host_to_device, int *hash_index, size_t data_step, in ParseDeviceData()
|
| D | device_embedding_operation.cc | 32 bool DeviceEmbeddingOperation::ParseHostDataHostToDevice(int id, size_t data_step, size_t *cur_grap… in ParseHostDataHostToDevice() 108 bool DeviceEmbeddingOperation::ParseHostDataDeviceToHost(size_t data_step, size_t *cur_graph_runnin… in ParseHostDataDeviceToHost()
|
| D | device_embedding_operation.h | 74 virtual bool AnalyseCache(int *batch_ids, const size_t batch_ids_num, size_t data_step, in AnalyseCache()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/distributed/embedding_cache/ |
| D | embedding_hash_map.cc | 52 … const size_t data_step, const size_t graph_running_step, size_t *const swap_out_size, in ParseData()
|