Searched refs:OptimizerKernel (Results 1 – 14 of 14) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/fl/server/kernel/ |
| D | optimizer_kernel_factory.h | 30 using OptimizerKernelCreator = std::function<std::shared_ptr<OptimizerKernel>()>;
31 class OptimizerKernelFactory : public KernelFactory<std::shared_ptr<OptimizerKernel>, OptimizerKern…
58 …static_assert(std::is_base_of<OptimizerKernel, CLASS<T>>::value, " must be base of OptimizerKernel…
|
| D | optimizer_kernel.h | 39 class OptimizerKernel : public CPUKernel {
41 OptimizerKernel() = default;
42 virtual ~OptimizerKernel() = default;
|
| D | apply_momentum_kernel.h | 33 class ApplyMomentumKernel : public ApplyMomentumCPUKernel, public OptimizerKernel {
|
| D | sgd_kernel.h | 33 class SGDKernel : public SGDCPUKernel<T>, public OptimizerKernel {
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32_grad/ |
| D | adam.h | 25 class AdamCPUKernel : public OptimizerKernel {
29 : OptimizerKernel(parameter, inputs, outputs, ctx, 5, 9), thread_count_(ctx->thread_num_) { in AdamCPUKernel()
|
| D | apply_momentum.h | 25 class ApplyMomentumCPUKernel : public OptimizerKernel {
29 : OptimizerKernel(parameter, inputs, outputs, ctx, 2, 3), in ApplyMomentumCPUKernel()
|
| D | sgd.h | 25 class SgdCPUKernel : public OptimizerKernel {
29 …: OptimizerKernel(parameter, inputs, outputs, ctx, 2, 1), thread_count_(ctx->thread_num_), sgd_par… in SgdCPUKernel()
|
| D | apply_momentum.cc | 106 auto ret = OptimizerKernel::Init(); in Init()
136 OptimizerKernel::OptimizerStep(); in OptimizerStep()
|
| D | adam.cc | 122 auto ret = OptimizerKernel::Init(); in Init()
158 OptimizerKernel::OptimizerStep(); in OptimizerStep()
|
| D | sgd.cc | 189 auto ret = OptimizerKernel::Init(); in Init()
224 OptimizerKernel::OptimizerStep(); in OptimizerStep()
|
| /third_party/mindspore/mindspore/lite/src/train/ |
| D | optimizer_kernel.h | 39 class OptimizerKernel : public InnerKernel {
41 OptimizerKernel() = default;
42 OptimizerKernel(OpParameter *parameter, const std::vector<lite::Tensor *> &inputs, in OptimizerKernel() function
45 ~OptimizerKernel() = default;
|
| D | train_session.cc | 652 auto optimizer = static_cast<kernel::OptimizerKernel *>(kernel->kernel()); in CompileOptimizedKernels()
677 auto optimizer = static_cast<kernel::OptimizerKernel *>(kernel->kernel()); in SetLearningRate()
691 auto optimizer = static_cast<kernel::OptimizerKernel *>(kernel->kernel()); in GetLearningRate()
702 auto optimizer = static_cast<kernel::OptimizerKernel *>(kernel->kernel()); in GetOptimizerParams()
733 auto optimizer = static_cast<kernel::OptimizerKernel *>(kernel->kernel()); in SetOptimizerParams()
751 auto optimizer = static_cast<kernel::OptimizerKernel *>(kernel->kernel()); in GetGradients()
797 auto optimizer = static_cast<kernel::OptimizerKernel *>(kernel->kernel()); in ApplyGradients()
820 auto optimizer = static_cast<kernel::OptimizerKernel *>(kernel->kernel()); in AdminSetupVirtualBatch()
871 auto optimizer = static_cast<kernel::OptimizerKernel *>(kernel->kernel()); in OptimizerStep()
|
| /third_party/mindspore/mindspore/ccsrc/fl/server/ |
| D | parameter_aggregator.h | 122 bool GenerateOptimizerKernelParams(const std::shared_ptr<kernel::OptimizerKernel> &optim_kernel,
148 …std::vector<std::pair<std::shared_ptr<kernel::OptimizerKernel>, KernelParams>> optimizer_kernel_pa…
|
| D | parameter_aggregator.cc | 117 std::shared_ptr<kernel::OptimizerKernel> optimizer_kernel = optimizer_with_params.first; in LaunchOptimizers()
327 const std::shared_ptr<kernel::OptimizerKernel> &optimizer_kernel, in GenerateOptimizerKernelParams()
387 template bool ParameterAggregator::AssignMemory(std::shared_ptr<kernel::OptimizerKernel> server_ker…
|