From 6a6813588da22be8d59c3dbdba71617947e7d890 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Fri, 18 Oct 2024 19:45:39 -0700 Subject: [PATCH] Enable mixed precision for VoxelMax. PiperOrigin-RevId: 687500971 --- tensorflow/core/grappler/optimizers/auto_mixed_precision_lists.h | 1 + 1 file changed, 1 insertion(+) diff --git a/tensorflow/core/grappler/optimizers/auto_mixed_precision_lists.h b/tensorflow/core/grappler/optimizers/auto_mixed_precision_lists.h index 5e39c80f9fd424..37f3714c4765b3 100644 --- a/tensorflow/core/grappler/optimizers/auto_mixed_precision_lists.h +++ b/tensorflow/core/grappler/optimizers/auto_mixed_precision_lists.h @@ -152,6 +152,7 @@ class AutoMixedPrecisionListsFp16 : public AutoMixedPrecisionLists { list.insert("TmlpV3"); list.insert("Pmlp"); list.insert("FastUnsortedSegmentMax"); + list.insert("VoxelMax"); } #if TENSORFLOW_USE_ROCM if (true) {