Searched refs:SparseApplyLazyAdamPSKernel (Results 1 – 3 of 3) sorted by relevance
28 void SparseApplyLazyAdamPSKernel::InitKernel( in InitKernel()82 void SparseApplyLazyAdamPSKernel::ReInit(const std::vector<std::vector<size_t>> &shapes) { in ReInit()92 void SparseApplyLazyAdamPSKernel::ReInit(const std::vector<AddressPtr> &inputs) { in ReInit()103 bool SparseApplyLazyAdamPSKernel::Execute(const std::vector<AddressPtr> &inputs, in Execute()113 const std::vector<size_t> &SparseApplyLazyAdamPSKernel::input_sizes() const { return GetInputSizeLi… in input_sizes()115 const std::vector<size_t> &SparseApplyLazyAdamPSKernel::output_sizes() const { return GetOutputSize… in output_sizes()117 const std::vector<size_t> &SparseApplyLazyAdamPSKernel::workspace_sizes() const { return GetWorkspa… in workspace_sizes()
28 class SparseApplyLazyAdamPSKernel : public SparseApplyLazyAdamCPUKernel, public PServerKernel {30 SparseApplyLazyAdamPSKernel(size_t rank_id, size_t pserver_num, size_t worker_num) in SparseApplyLazyAdamPSKernel() function32 ~SparseApplyLazyAdamPSKernel() override = default;
130 …std::make_shared<kernel::ps::SparseApplyLazyAdamPSKernel>(server_node_->rank_id(), pserver_num_, w… in InitOptimInputsShape()