tf.keras.dtype_policies.get

Retrieves a Keras DTypePolicy instance.

tf.keras.dtype_policies.get(
    identifier
)

The identifier may be the string name of a DTypePolicy class.

policy = dtype_policies.get("mixed_bfloat16")
type(loss)
<class '...FloatDTypePolicy'>

You can also specify config of the dtype policy to this function by passing dict containing class_name and config as an identifier. Also note that the class_name must map to a DTypePolicy class

identifier = {"class_name": "FloatDTypePolicy",
              "config": {"name": "float32"} }
policy = dtype_policies.get(identifier)
type(loss)
<class '...FloatDTypePolicy'>

identifier A dtype policy identifier. One of None or string name of a DTypePolicy or DTypePolicy configuration dictionary or a DTypePolicy instance.

A Keras DTypePolicy instance.