Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dmixed_precision_graph_rewrite_test.py29 from tensorflow.python.keras.optimizer_v2 import gradient_descent as gradient_descent_v2 unknown
65 opt = gradient_descent_v2.SGD(1.0)
73 opt = gradient_descent_v2.SGD(1.0)
87 enable_mixed_precision_graph_rewrite(gradient_descent_v2.SGD(1.0))
89 enable_mixed_precision_graph_rewrite(gradient_descent_v2.SGD(1.0))
92 enable_mixed_precision_graph_rewrite(gradient_descent_v2.SGD(1.0))
Dautocast_variable_test.py45 from tensorflow.python.keras.optimizer_v2 import gradient_descent as gradient_descent_v2 unknown
535 gradient_descent_v2.SGD,
569 if optimizer_class in (gradient_descent_v2.SGD,
/external/tensorflow/tensorflow/python/keras/
Doptimizers.py34 from tensorflow.python.keras.optimizer_v2 import gradient_descent as gradient_descent_v2 unknown
71 'sgd': gradient_descent_v2.SGD,
Dtesting_utils.py46 from tensorflow.python.keras.optimizer_v2 import gradient_descent as gradient_descent_v2 unknown
889 'sgd': gradient_descent_v2.SGD
/external/tensorflow/tensorflow/python/keras/saving/
Dsave_test.py798 optimizer=optimizers.gradient_descent_v2.SGD(),