Searched refs:update_vals (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ps/ |
| D | embedding_look_up_ps_kernel.cc | 91 const float *update_vals, size_t ids_size) { in UpdateEmbeddings() argument
100 update_vals + i * outer_dim_size_, copy_len); in UpdateEmbeddings()
|
| D | embedding_look_up_ps_kernel.h | 39 void UpdateEmbeddings(float *embedding_table, const size_t *lookup_ids, const float *update_vals,
|
| D | pserver_kernel.h | 41 …l void UpdateEmbeddings(float *embedding_table, const size_t *lookup_ids, const float *update_vals, in UpdateEmbeddings() argument
|
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | parameter_server.cc | 818 const Values &update_vals = {input.values().begin(), input.values().end()}; in HandleUpdateEmbeddings() local
819 ps_->UpdateEmbeddings(key, lookup_ids, update_vals); in HandleUpdateEmbeddings()
|