Searched refs:pserver_num_ (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ps/ |
| D | pserver_kernel.h | 31 : rank_id_(rank_id), pserver_num_(pserver_num), worker_num_(worker_num) {} in PServerKernel()
52 size_t pserver_num_; variable
|
| D | pserver_kernel.cc | 28 …LocalShard(SizeToLong((*shape)[IntToSize(axis)]), SizeToLong(rank_id_), SizeToLong(pserver_num_))); in Shard()
|
| D | embedding_look_up_ps_kernel.cc | 58 …fset += Util::LocalShard(SizeToLong(input_shape_[kAxis]), SizeToLong(i), SizeToLong(pserver_num_)); in InitKernel()
|
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | parameter_server.h | 79 : pserver_num_(0), in ParameterServer()
148 size_t pserver_num_; variable
|
| D | parameter_server.cc | 50 pserver_num_ = std::strtol(mindspore::common::GetEnv(kEnvPServerNum).c_str(), nullptr, kBase); in Init()
125 …ke_shared<kernel::ps::SparseApplyAdamPSKernel>(server_node_->rank_id(), pserver_num_, worker_num_); in InitOptimInputsShape()
130 …hared<kernel::ps::SparseApplyLazyAdamPSKernel>(server_node_->rank_id(), pserver_num_, worker_num_); in InitOptimInputsShape()
135 …std::make_shared<kernel::ps::ApplyMomentumPSKernel>(server_node_->rank_id(), pserver_num_, worker_… in InitOptimInputsShape()
140 …ke_shared<kernel::ps::SparseApplyFtrlPSKernel>(server_node_->rank_id(), pserver_num_, worker_num_); in InitOptimInputsShape()
222 …std::make_shared<kernel::ps::EmbeddingLookUpPSKernel>(server_node_->rank_id(), pserver_num_, worke… in InitEmbeddingTable()
310 optim_info->ComputeMean(shapes, worker_num_, pserver_num_, server_node_->rank_id()); in UpdateWeights()
|