Searched refs:OptimizationParameters (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_embedding_optimization_parameters_utils.h | 29 using OptimizationAlgorithm = OptimizationParameters::ParametersCase;
51 Status GetBaseAuxiliaryParameterCount(const OptimizationParameters ¶ms,
56 Status GetGradientAccumulationSupport(const OptimizationParameters ¶ms,
63 const OptimizationParameters ¶ms, bool use_gradient_accumulation,
|
| D | tpu_embedding_optimization_parameters_utils.cc | 108 Status GetBaseAuxiliaryParameterCount(const OptimizationParameters& params, in GetBaseAuxiliaryParameterCount()
179 Status GetGradientAccumulationSupport(const OptimizationParameters& params, in GetGradientAccumulationSupport()
206 const OptimizationParameters& params, bool use_gradient_accumulation, in GetOptimizationAlgorithmStateVariables()
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_embedding_v2_utils.py | 106 self, parameters: optimization_parameters_pb2.OptimizationParameters): argument
258 self, parameters: optimization_parameters_pb2.OptimizationParameters): argument
369 self, parameters: optimization_parameters_pb2.OptimizationParameters): argument
515 self, parameters: optimization_parameters_pb2.OptimizationParameters): argument
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | tpu_embedding_load_retrieve_ops.cc | 29 using OptimizationAlgorithm = OptimizationParameters::ParametersCase;
|
| /external/tensorflow/tensorflow/core/protobuf/tpu/ |
| D | tpu_embedding_configuration.proto | 21 OptimizationParameters optimization_parameters = 5;
|
| D | optimization_parameters.proto | 375 message OptimizationParameters { message
|