Searched defs:outer_dim_size (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | embedding_look_up_cpu_kernel.cc | 34 size_t outer_dim_size, T offset, size_t first_dim_size) { in LookUpTableTask()
|
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | util.cc | 109 const size_t first_dim_size, const size_t outer_dim_size, in ReduceSparseGradient()
|
| D | worker.cc | 113 …int64_t outer_dim_size = std::accumulate(var_shape.begin() + 1, var_shape.end(), 1, std::multiplie… in Push() local
660 … size_t grad_index, size_t indice_index, size_t first_dim_size, size_t outer_dim_size) { in PushSparseData()
739 size_t outer_dim_size = static_cast<size_t>(iter->second); in SparsePartitioner() local
|
| D | optimizer_info.cc | 202 size_t outer_dim_size = segment_size; in ComputeMean() local
|
| /third_party/mindspore/mindspore/ccsrc/ps/ps_cache/ |
| D | ps_cache_manager.cc | 682 void PsCacheManager::LookUpTableTask(size_t indices_lens, size_t outer_dim_size, size_t first_dim_s… in LookUpTableTask()
717 size_t outer_dim_size = embedding_size; in LookUpHostHashTable() local
758 … const int *insert_indices, const float *insert_data, float *hash_table_addr) { in InsertHostHashTable()
|