Searched refs:SetOptimizerMode (Results 1 – 2 of 2) sorted by relevance
124 int SetOptimizerMode(WeightUpdateMode mod) { in SetOptimizerMode() function
653 optimizer->SetOptimizerMode(kernel::WeightUpdateMode::ACCUMULATE_GRADS); in CompileOptimizedKernels()826 auto ret = optimizer->SetOptimizerMode(mod); in AdminSetupVirtualBatch()