Home
last modified time | relevance | path

Searched defs:set_weights (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/keras/optimizer_v2/
H A Dadam.py154 def set_weights(self, weights): member in Adam
407 def set_weights(self, weights): member in NonFusedAdam
H A Dadadelta.py97 def set_weights(self, weights): member in Adadelta
H A Drmsprop.py277 def set_weights(self, weights): member in RMSprop
H A Dadagrad.py100 def set_weights(self, weights): member in Adagrad
H A Doptimizer_v2.py1113 def set_weights(self, weights): member in OptimizerV2
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/engine/
H A Dbase_layer_utils.py847 def set_weights(self, weights): member in TrackableWeightHandler
H A Dbase_layer_v1.py1225 def set_weights(self, weights): member in Layer
H A Dbase_layer.py1723 def set_weights(self, weights): member in Layer
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/
H A Doptimizer_v1.py110 def set_weights(self, weights): member in Optimizer
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/distribute/
H A Ddistributed_training_utils_v1.py51 def set_weights(distribution_strategy, dist_model, weights): function
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/mixed_precision/
H A Dloss_scale_optimizer.py755 def set_weights(self, weights): member in LossScaleOptimizer