Searched refs:first_dim_size (Results 1 – 8 of 8) 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() argument
39 if (index >= 0 && index < SizeToInt(first_dim_size)) { in LookUpTableTask()
|
| /third_party/mindspore/mindspore/ccsrc/ps/ps_cache/ |
| D | ps_cache_manager.cc | 682 … PsCacheManager::LookUpTableTask(size_t indices_lens, size_t outer_dim_size, size_t first_dim_size, in LookUpTableTask() argument
691 if (index >= 0 && index < SizeToInt(first_dim_size)) { in LookUpTableTask()
716 size_t first_dim_size = host_vocab_cache_size_; in LookUpHostHashTable() local
731 …std::thread(&PsCacheManager::LookUpTableTask, this, task_proc_lens, outer_dim_size, first_dim_size, in LookUpHostHashTable()
749 size_t first_dim_size = host_vocab_cache_size_; in InsertHostHashTable() local
757 …_hash_table_task = [this](size_t insert_indices_size, size_t outer_dim_size, size_t first_dim_size, in InsertHostHashTable()
764 if (index >= 0 && index < SizeToInt(first_dim_size)) { in InsertHostHashTable()
781 threads[i] = std::thread(insert_hash_table_task, task_proc_lens, embedding_size, first_dim_size, in InsertHostHashTable()
|
| D | ps_cache_manager.h | 170 …void LookUpTableTask(size_t indices_lens, size_t outer_dim_size, size_t first_dim_size, const floa…
|
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | util.h | 55 const size_t first_dim_size, const size_t outer_dim_size,
|
| D | optimizer_info.cc | 201 size_t first_dim_size = input_shapes.front(); in ComputeMean() local
204 if (first_dim_size == 0 || outer_dim_size == 0) { in ComputeMean()
231 …Util::ReduceSparseGradient(grad_data, indices_data, indices_size, segment_size, first_dim_size, ou… in ComputeMean()
|
| D | util.cc | 109 const size_t first_dim_size, const size_t outer_dim_size, in ReduceSparseGradient() argument
125 param.max_index_ = first_dim_size; in ReduceSparseGradient()
|
| D | worker.cc | 112 int64_t first_dim_size = var_shape[0]; in Push() local
116 … << " the first_dim_size:" << first_dim_size << " the outer_dim_size" << outer_dim_size; in Push()
118 … LongToSize(indice_index), LongToSize(first_dim_size), LongToSize(outer_dim_size)); in Push()
660 … size_t grad_index, size_t indice_index, size_t first_dim_size, size_t outer_dim_size) { in PushSparseData() argument
666 …std::map<int64_t, int64_t> attrs{{0, grad_index}, {1, indice_index}, {2, first_dim_size}, {3, oute… in PushSparseData()
737 size_t first_dim_size = static_cast<size_t>(iter->second); in SparsePartitioner() local
802 first_dim_size, outer_dim_size, &unique_sparse_grad); in SparsePartitioner()
|
| D | worker.h | 107 … size_t grad_index, size_t indice_index, size_t first_dim_size, size_t outer_dim_size);
|