Searched refs:add_state_variable (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_embedding_optimization_parameters_utils.cc | 212 auto add_state_variable = [&](const std::string& name, float value) { in GetOptimizationAlgorithmStateVariables() local
218 add_state_variable("parameters", 0.0); in GetOptimizationAlgorithmStateVariables()
219 add_state_variable("accumulators", 0.1); in GetOptimizationAlgorithmStateVariables()
223 add_state_variable("parameters", 0.0); in GetOptimizationAlgorithmStateVariables()
224 add_state_variable("accumulators", 0.1); in GetOptimizationAlgorithmStateVariables()
228 add_state_variable("parameters", 0.0); in GetOptimizationAlgorithmStateVariables()
232 add_state_variable("parameters", 0.0); in GetOptimizationAlgorithmStateVariables()
233 add_state_variable("accumulators", 0.1); in GetOptimizationAlgorithmStateVariables()
234 add_state_variable("linears", 0.0); in GetOptimizationAlgorithmStateVariables()
238 add_state_variable("parameters", 0.0); in GetOptimizationAlgorithmStateVariables()
[all …]
|