Home
last modified time | relevance | path

Searched refs:SetOptimizerMode (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/lite/src/train/
Doptimizer_kernel.h124 int SetOptimizerMode(WeightUpdateMode mod) { in SetOptimizerMode() function
Dtrain_session.cc653 optimizer->SetOptimizerMode(kernel::WeightUpdateMode::ACCUMULATE_GRADS); in CompileOptimizedKernels()
826 auto ret = optimizer->SetOptimizerMode(mod); in AdminSetupVirtualBatch()