Home
last modified time | relevance | path

Searched refs:pserver_kernel (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/ps/
Doptimizer_info_builder.h36 …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() argument
30 MS_EXCEPTION_IF_NULL(pserver_kernel); in Build()
34 BuildInputs(weight, keys, values, lens, inputs_shape, worker_num, pserver_kernel, sharded); in Build()
36 std::vector<size_t> ws_sizes = pserver_kernel->workspace_sizes(); in Build()
207 … const std::shared_ptr<PServerKernel> &pserver_kernel, in BuildInputs() argument
223 tmp[i] = std::dynamic_pointer_cast<SparseApplyFtrlPSKernel>(pserver_kernel)->init_accum(); in BuildInputs()
Dparameter_server.cc332 std::shared_ptr<kernel::ps::PServerKernel> pserver_kernel = optimizers_[key]; in AccumGrad() local
333 if (pserver_kernel == nullptr) { in AccumGrad()
336 MS_EXCEPTION_IF_NULL(pserver_kernel); in AccumGrad()
337 OptimizerInfo *optim = builder->Build(pserver_kernel, weights_[key], keys, values, lengths, in AccumGrad()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/
DCMakeLists.txt56 list(REMOVE_ITEM CPU_SRC_LIST "cpu/ps/pserver_kernel.cc")