tf.keras.mixed_precision.experimental.get_layer_policy

View source on GitHub

Returns the dtype policy of a layer.

tf.keras.mixed_precision.experimental.get_layer_policy(
    layer
)

Args:

Returns:

The tf.keras.mixed_precision.experimental.Policy of the layer.