Home
last modified time | relevance | path

Searched refs:OptimizationAlgorithm (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/
Dtpu_embedding_optimization_parameters_utils.cc29 string GetOptimizationAlgorithmName(OptimizationAlgorithm alg) { in GetOptimizationAlgorithmName()
31 case OptimizationAlgorithm::kAdagrad: in GetOptimizationAlgorithmName()
33 case OptimizationAlgorithm::kBoundedAdagrad: in GetOptimizationAlgorithmName()
35 case OptimizationAlgorithm::kStochasticGradientDescent: in GetOptimizationAlgorithmName()
37 case OptimizationAlgorithm::kFtrl: in GetOptimizationAlgorithmName()
39 case OptimizationAlgorithm::kAdam: in GetOptimizationAlgorithmName()
41 case OptimizationAlgorithm::kMomentum: in GetOptimizationAlgorithmName()
43 case OptimizationAlgorithm::kRmsProp: in GetOptimizationAlgorithmName()
45 case OptimizationAlgorithm::kCenteredRmsProp: in GetOptimizationAlgorithmName()
47 case OptimizationAlgorithm::kMdlAdagradLight: in GetOptimizationAlgorithmName()
[all …]
Dtpu_embedding_optimization_parameters_utils.h29 using OptimizationAlgorithm = OptimizationParameters::ParametersCase; variable
32 string GetOptimizationAlgorithmName(OptimizationAlgorithm alg);
35 string GetOptimizationAlgorithmFriendlyName(OptimizationAlgorithm alg);
38 std::vector<OptimizationAlgorithm> GetOptimizationAlgorithms();
/external/tensorflow/tensorflow/core/ops/
Dtpu_embedding_load_retrieve_ops.cc29 using OptimizationAlgorithm = OptimizationParameters::ParametersCase; typedef