def METHOD_NAME(
    self, distribution, use_numpy, use_validation_data
):
    if isinstance(
        distribution,
        (
            tf.distribute.experimental.CentralStorageStrategy,
            tf.compat.v1.distribute.experimental.CentralStorageStrategy,
        ),
    ):
        self.skipTest(
            "CentralStorageStrategy is not supported by mixed precision."
        )
    if isinstance(
        distribution,
        (
            tf.distribute.experimental.TPUStrategy,
            tf.compat.v1.distribute.experimental.TPUStrategy,
        ),
    ):
        policy_name = "mixed_bfloat16"
    else:
        policy_name = "mixed_float16"
    with policy.policy_scope(policy_name):
        self.run_correctness_test(
            distribution, use_numpy, use_validation_data
        )