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'>

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

Returns
A Keras `DTypePolicy` instance.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.

Last updated 2024-06-07 UTC.

tf.keras.dtype_policies.get

Args

Returns