From fa3d7213026e78f158d49c4e998d885068b51b9c Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Fri, 29 May 2026 11:28:59 -0700 Subject: [PATCH] Add float6_e3m2fn and float6_e2m3fn aliases to tsl/platform/ml_dtypes.h. PiperOrigin-RevId: 923516316 --- tsl/platform/ml_dtypes.h | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tsl/platform/ml_dtypes.h b/tsl/platform/ml_dtypes.h index b536e2f8a..b035546ca 100644 --- a/tsl/platform/ml_dtypes.h +++ b/tsl/platform/ml_dtypes.h @@ -30,6 +30,8 @@ using float8_e4m3b11fnuz = ::ml_dtypes::float8_e4m3b11fnuz; using float8_e5m2 = ::ml_dtypes::float8_e5m2; using float8_e5m2fnuz = ::ml_dtypes::float8_e5m2fnuz; using float8_e8m0fnu = ::ml_dtypes::float8_e8m0fnu; +using float6_e3m2fn = ::ml_dtypes::float6_e3m2fn; +using float6_e2m3fn = ::ml_dtypes::float6_e2m3fn; using int1 = ::ml_dtypes::int1; using uint1 = ::ml_dtypes::uint1;