Searched refs:OptimizationAlgorithm (Results 1 – 4 of 4) sorted by relevance
32 std::string GetOptimizationAlgorithmName(OptimizationAlgorithm alg) { in GetOptimizationAlgorithmName()34 case OptimizationAlgorithm::kAdagrad: in GetOptimizationAlgorithmName()36 case OptimizationAlgorithm::kAdagradMomentum: in GetOptimizationAlgorithmName()38 case OptimizationAlgorithm::kBoundedAdagrad: in GetOptimizationAlgorithmName()40 case OptimizationAlgorithm::kStochasticGradientDescent: in GetOptimizationAlgorithmName()42 case OptimizationAlgorithm::kFtrl: in GetOptimizationAlgorithmName()44 case OptimizationAlgorithm::kAdam: in GetOptimizationAlgorithmName()46 case OptimizationAlgorithm::kMomentum: in GetOptimizationAlgorithmName()48 case OptimizationAlgorithm::kRmsProp: in GetOptimizationAlgorithmName()50 case OptimizationAlgorithm::kCenteredRmsProp: in GetOptimizationAlgorithmName()[all …]
29 using OptimizationAlgorithm = OptimizationParameters::ParametersCase; variable32 string GetOptimizationAlgorithmName(OptimizationAlgorithm alg);35 string GetOptimizationAlgorithmFriendlyName(OptimizationAlgorithm alg);38 std::vector<OptimizationAlgorithm> GetOptimizationAlgorithms();
178 for (tpu::OptimizationAlgorithm alg : tpu::GetOptimizationAlgorithms()) { in GetPerTableLoadOptimizationParametersOps()197 for (tpu::OptimizationAlgorithm alg : tpu::GetOptimizationAlgorithms()) { in GetPerTableRetrieveOptimizationParametersOps()
29 using OptimizationAlgorithm = OptimizationParameters::ParametersCase; typedef