Searched defs:first_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/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()
716 size_t first_dim_size = host_vocab_cache_size_; in LookUpHostHashTable() local
749 size_t first_dim_size = host_vocab_cache_size_; in InsertHostHashTable() local
758 … const int *insert_indices, const float *insert_data, float *hash_table_addr) { in InsertHostHashTable()
|
| /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 | 112 int64_t first_dim_size = var_shape[0]; in Push() local
660 … size_t grad_index, size_t indice_index, size_t first_dim_size, size_t outer_dim_size) { in PushSparseData()
737 size_t first_dim_size = static_cast<size_t>(iter->second); in SparsePartitioner() local
|
| D | optimizer_info.cc | 201 size_t first_dim_size = input_shapes.front(); in ComputeMean() local
|