Searched defs:OptimizerStep (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32_grad/ | ||
| D | apply_momentum.cc | 119 int ApplyMomentumCPUKernel::OptimizerStep() { in OptimizerStep() function in mindspore::kernel::ApplyMomentumCPUKernel |
| D | adam.cc | 135 int AdamCPUKernel::OptimizerStep() { in OptimizerStep() function in mindspore::kernel::AdamCPUKernel |
| D | sgd.cc | 202 int SgdCPUKernel::OptimizerStep() { in OptimizerStep() function in mindspore::kernel::SgdCPUKernel |
| /third_party/mindspore/mindspore/lite/src/train/ | ||
| D | optimizer_kernel.h | 165 virtual int OptimizerStep() { in OptimizerStep() function |
| D | train_session.cc | 868 int TrainSession::OptimizerStep() { in OptimizerStep() function in mindspore::lite::TrainSession |