Enable mixed precision for VoxelMax.

PiperOrigin-RevId: 687500971
Intel-tensorflow · Oct 19, 2024 · 6a68135 · 6a68135
1 parent d8c3664
commit 6a68135
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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) {