Home
last modified time | relevance | path

Searched defs:ReInit (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ps/
Dpserver_kernel.h38 virtual void ReInit(const std::vector<std::vector<size_t>> &) {} in ReInit() function
48 virtual void ReInit(const std::vector<AddressPtr> &) {} in ReInit() function
Dsparse_apply_ftrl_ps_kernel.cc92 void SparseApplyFtrlPSKernel::ReInit(const std::vector<std::vector<size_t>> &shapes) { in ReInit() function in mindspore::kernel::ps::SparseApplyFtrlPSKernel
102 void SparseApplyFtrlPSKernel::ReInit(const std::vector<AddressPtr> &inputs) { in ReInit() function in mindspore::kernel::ps::SparseApplyFtrlPSKernel
Dsparse_apply_adam_ps_kernel.cc81 void SparseApplyAdamPSKernel::ReInit(const std::vector<std::vector<size_t>> &shapes) { in ReInit() function in mindspore::kernel::ps::SparseApplyAdamPSKernel
91 void SparseApplyAdamPSKernel::ReInit(const std::vector<AddressPtr> &inputs) { in ReInit() function in mindspore::kernel::ps::SparseApplyAdamPSKernel
Dsparse_apply_lazy_adam_ps_kernel.cc82 void SparseApplyLazyAdamPSKernel::ReInit(const std::vector<std::vector<size_t>> &shapes) { in ReInit() function in mindspore::kernel::ps::SparseApplyLazyAdamPSKernel
92 void SparseApplyLazyAdamPSKernel::ReInit(const std::vector<AddressPtr> &inputs) { in ReInit() function in mindspore::kernel::ps::SparseApplyLazyAdamPSKernel
Dembedding_look_up_ps_kernel.cc70 void EmbeddingLookUpPSKernel::ReInit(const std::vector<std::vector<size_t>> &shapes) { in ReInit() function in mindspore::kernel::ps::EmbeddingLookUpPSKernel