Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/tpu/
Dtpu_embedding_optimization_parameters_utils.cc320 StateVariableSpecification gradient_acc; in GetOptimizationAlgorithmStateVariables() local
321 gradient_acc.set_name("gradient_accumulators"); in GetOptimizationAlgorithmStateVariables()
322 gradient_acc.mutable_fill_with_constant()->set_initial_value( in GetOptimizationAlgorithmStateVariables()
324 state_variables->push_back(std::move(gradient_acc)); in GetOptimizationAlgorithmStateVariables()