Searched refs:optimization_parameters (Results 1 – 7 of 7) sorted by relevance
74 optimization_parameters=None): argument138 if optimization_parameters is not None:139 if not isinstance(optimization_parameters, _OptimizationParameters):143 type(optimization_parameters)))148 learning_rate_fn, optimization_parameters)1227 optimization_parameters=None, argument1346 _validate_optimization_parameters(optimization_parameters,1348 self._optimization_parameters = optimization_parameters1350 if optimization_parameters is not None:1442 def optimization_parameters(self): member in TPUEmbedding[all …]
489 parameters = table_descriptor.optimization_parameters
23 name: "optimization_parameters"
20 //third_party/tensorflow/core/protobuf/tpu/optimization_parameters.proto.
28 "optimization_parameters.proto",
5 import "tensorflow/core/protobuf/tpu/optimization_parameters.proto";21 OptimizationParameters optimization_parameters = 5; field
937 //third_party/tensorflow/core/protobuf/tpu/optimization_parameters.proto.