tf.keras.backend.set_floatx

tf.keras.backend.set_floatx(value)

Defined in tensorflow/python/keras/_impl/keras/backend.py.

Sets the default float type.

Arguments:

• value: String; 'float16', 'float32', or 'float64'.

Example:

>>> from keras import backend as K
>>> K.floatx()
'float32'
>>> K.set_floatx('float16')
>>> K.floatx()
'float16'

Raises:

• ValueError: In case of invalid value.