Searched refs:pserver_num (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ps/ |
| D | apply_momentum_ps_kernel.h | 30 ApplyMomentumPSKernel(size_t rank_id, size_t pserver_num, size_t worker_num) in ApplyMomentumPSKernel() argument
31 : PServerKernel(rank_id, pserver_num, worker_num) {} in ApplyMomentumPSKernel()
|
| D | sparse_apply_adam_ps_kernel.h | 31 SparseApplyAdamPSKernel(size_t rank_id, size_t pserver_num, size_t worker_num) in SparseApplyAdamPSKernel() argument
32 : PServerKernel(rank_id, pserver_num, worker_num) {} in SparseApplyAdamPSKernel()
|
| D | sparse_apply_lazy_adam_ps_kernel.h | 30 SparseApplyLazyAdamPSKernel(size_t rank_id, size_t pserver_num, size_t worker_num) in SparseApplyLazyAdamPSKernel() argument
31 : PServerKernel(rank_id, pserver_num, worker_num) {} in SparseApplyLazyAdamPSKernel()
|
| D | embedding_look_up_ps_kernel.h | 30 EmbeddingLookUpPSKernel(size_t rank_id, size_t pserver_num, size_t worker_num) in EmbeddingLookUpPSKernel() argument
31 : PServerKernel(rank_id, pserver_num, worker_num) {} in EmbeddingLookUpPSKernel()
|
| D | sparse_apply_ftrl_ps_kernel.h | 31 SparseApplyFtrlPSKernel(size_t rank_id, size_t pserver_num, size_t worker_num) in SparseApplyFtrlPSKernel() argument
32 : PServerKernel(rank_id, pserver_num, worker_num), init_accum_(0.1) {} in SparseApplyFtrlPSKernel()
|
| D | pserver_kernel.h | 30 PServerKernel(size_t rank_id, size_t pserver_num, size_t worker_num) in PServerKernel() argument
31 : rank_id_(rank_id), pserver_num_(pserver_num), worker_num_(worker_num) {} in PServerKernel()
|