Home
last modified time | relevance | path

Searched refs:PServerKernel (Results 1 – 11 of 11) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ps/
Dpserver_kernel.h28 class PServerKernel {
30 PServerKernel(size_t rank_id, size_t pserver_num, size_t worker_num) in PServerKernel() function
32 ~PServerKernel() = default;
33 PServerKernel(const PServerKernel &) = delete;
34 PServerKernel &operator=(const PServerKernel &) = delete;
Dapply_momentum_ps_kernel.h28 class ApplyMomentumPSKernel : public ApplyMomentumCPUKernel, public PServerKernel {
31 : PServerKernel(rank_id, pserver_num, worker_num) {} in ApplyMomentumPSKernel()
Dsparse_apply_adam_ps_kernel.h29 class SparseApplyAdamPSKernel : public SparseApplyAdamCPUKernel, public PServerKernel {
32 : PServerKernel(rank_id, pserver_num, worker_num) {} in SparseApplyAdamPSKernel()
Dsparse_apply_lazy_adam_ps_kernel.h28 class SparseApplyLazyAdamPSKernel : public SparseApplyLazyAdamCPUKernel, public PServerKernel {
31 : PServerKernel(rank_id, pserver_num, worker_num) {} in SparseApplyLazyAdamPSKernel()
Dembedding_look_up_ps_kernel.h28 class EmbeddingLookUpPSKernel : public EmbeddingLookUpCPUKernel, public PServerKernel {
31 : PServerKernel(rank_id, pserver_num, worker_num) {} in EmbeddingLookUpPSKernel()
Dsparse_apply_ftrl_ps_kernel.h29 class SparseApplyFtrlPSKernel : public SparseApplyFtrlCPUKernel, public PServerKernel {
32 : PServerKernel(rank_id, pserver_num, worker_num), init_accum_(0.1) {} in SparseApplyFtrlPSKernel()
Dpserver_kernel.cc22 void PServerKernel::Shard(std::vector<size_t> *shape, int axis) { in Shard()
/third_party/mindspore/mindspore/ccsrc/ps/
Doptimizer_info_builder.h30 using mindspore::kernel::ps::PServerKernel;
36 …OptimizerInfo *Build(const std::shared_ptr<PServerKernel> &pserver_kernel, const WeightPtr &weight…
42 … const std::shared_ptr<PServerKernel> &pserver_kernel, bool sharded) = 0;
61 … const std::shared_ptr<PServerKernel> &pserver_kernel, bool sharded) override;
70 … const std::shared_ptr<PServerKernel> &pserver_kernel, bool sharded) override;
79 … const std::shared_ptr<PServerKernel> &pserver_kernel, bool sharded) override;
Doptimizer_info_builder.cc26 OptimizerInfo *OptimizerInfoBuilder::Build(const std::shared_ptr<PServerKernel> &pserver_kernel, in Build()
116 const std::shared_ptr<PServerKernel> &, bool) { in BuildInputs() argument
148 … const std::shared_ptr<PServerKernel> &, bool sharded) { in BuildInputs() argument
207 … const std::shared_ptr<PServerKernel> &pserver_kernel, in BuildInputs()
Dparameter_server.cc124 std::shared_ptr<PServerKernel> optimizer = in InitOptimInputsShape()
129 std::shared_ptr<PServerKernel> optimizer = in InitOptimInputsShape()
134 std::shared_ptr<PServerKernel> optimizer = in InitOptimInputsShape()
139 std::shared_ptr<PServerKernel> optimizer = in InitOptimInputsShape()
221 std::shared_ptr<PServerKernel> lookup = in InitEmbeddingTable()
285 std::shared_ptr<PServerKernel> optimizer = nullptr; in UpdateWeights()
332 std::shared_ptr<kernel::ps::PServerKernel> pserver_kernel = optimizers_[key]; in AccumGrad()
383 std::shared_ptr<PServerKernel> table_lookup_op = embedding_lookup_ops_[key]; in DoEmbeddingLookup()
439 std::shared_ptr<PServerKernel> table_lookup_op = embedding_lookup_ops_[key]; in UpdateEmbeddings()
Dparameter_server.h159 std::unordered_map<Key, std::shared_ptr<PServerKernel>> optimizers_;
170 std::unordered_map<Key, std::shared_ptr<PServerKernel>> embedding_lookup_ops_;