Searched refs:optimization_parameters (Results 1 – 12 of 12) sorted by relevance
67 optimization_parameters=None): argument135 if optimization_parameters is not None:136 if not isinstance(optimization_parameters, _OptimizationParameters):144 learning_rate_fn, optimization_parameters)1307 optimization_parameters=None, argument1426 _validate_optimization_parameters(optimization_parameters,1428 self._optimization_parameters = optimization_parameters1430 if optimization_parameters is not None:1523 def optimization_parameters(self): member in TPUEmbedding1540 optimization_parameters = ([all …]
569 parameters = table_descriptor.optimization_parameters
236 .optimization_parameters() in GetLoadOrRetrieveNodesByTable()296 .optimization_parameters(); in CombinePerTableParametersForLoad()451 .optimization_parameters(); in Run()
266 config.table_descriptor(table_id).optimization_parameters(), in __anoncc2b8c080702()337 config.table_descriptor(table_id).optimization_parameters(), in __anoncc2b8c080802()
31 table_descriptor.optimization_parameters().learning_rate(); in ComputeTotalTagCountForDynamicLearningRates()
148 table_descriptor.optimization_parameters(), &state_variable_specs); in GetStateVariables()299 table_descriptor.optimization_parameters(), &state_variable_specs); in GetStateVariables()
5 import "tensorflow/core/protobuf/tpu/optimization_parameters.proto";24 OptimizationParameters optimization_parameters = 5; field
27 "optimization_parameters.proto",
23 name: "optimization_parameters"
20 //third_party/tensorflow/core/protobuf/tpu/optimization_parameters.proto.
427 StrAttr:$optimization_parameters
14943 //third_party/tensorflow/core/protobuf/tpu/optimization_parameters.proto.