Merge pull request #18233 from tilakrayal:patch-8

PiperOrigin-RevId: 541932649
keras-team · Jun 20, 2023 · 5f9d052 · 5f9d052
2 parents 8e63b2f + 1c36702
commit 5f9d052
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/keras/mixed_precision/policy.py b/keras/mixed_precision/policy.py
@@ -198,7 +198,7 @@ def __init__(self, name):
             raise TypeError(f"'name' must be a string, but got: {name}")
         self._name = name
         self._compute_dtype, self._variable_dtype = self._parse_name(name)
-        if name in ("mixed_float16", "mixed_bloat16"):
+        if name in ("mixed_float16", "mixed_bfloat16"):
             device_compatibility_check.log_device_compatibility_check(name)
 
     def _parse_name(self, name):