Home
last modified time | relevance | path

Searched refs:enable_auto_cast_variables (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/keras/engine/
H A Dbase_layer_v1.py782 with autocast_variable.enable_auto_cast_variables(
827 with autocast_variable.enable_auto_cast_variables(
1033 with autocast_variable.enable_auto_cast_variables(None):
H A Dbase_layer.py887 with autocast_variable.enable_auto_cast_variables(
1042 with autocast_variable.enable_auto_cast_variables(
1482 with autocast_variable.enable_auto_cast_variables(None):
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/mixed_precision/
H A Dautocast_variable.py533 class enable_auto_cast_variables(object): # pylint:disable=invalid-name class
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/saving/saved_model/
H A Dsave_impl.py597 with autocast_variable.enable_auto_cast_variables(