Searched refs:enable_auto_cast_variables (Results 1 – 5 of 5) sorted by relevance
86 with autocast_variable.enable_auto_cast_variables(dtypes.float16):93 with autocast_variable.enable_auto_cast_variables(dtypes.float32):109 with autocast_variable.enable_auto_cast_variables(dtypes.float16):120 with autocast_variable.enable_auto_cast_variables(dtypes.float16):123 with autocast_variable.enable_auto_cast_variables(dtypes.float32):139 with autocast_variable.enable_auto_cast_variables(dtype):165 with autocast_variable.enable_auto_cast_variables(read_dtype):198 with autocast_variable.enable_auto_cast_variables(read_dtype):229 with autocast_variable.enable_auto_cast_variables(read_dtype):341 with autocast_variable.enable_auto_cast_variables(dtypes.float16):[all …]
544 class enable_auto_cast_variables(object): # pylint:disable=invalid-name class
782 with autocast_variable.enable_auto_cast_variables(827 with autocast_variable.enable_auto_cast_variables(1033 with autocast_variable.enable_auto_cast_variables(None):
876 with autocast_variable.enable_auto_cast_variables(1030 with autocast_variable.enable_auto_cast_variables(1164 with autocast_variable.enable_auto_cast_variables(1563 with autocast_variable.enable_auto_cast_variables(None):
524 with autocast_variable.enable_auto_cast_variables(