Enable mixed precision for VoxelMax.

tensorflower-gardener · tensorflower-gardener · commit 6a6813588da2 · 2024-10-18T19:54:05.000-07:00
PiperOrigin-RevId: 687500971
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) {

Original file line number	Diff line number	Diff line change
`@@ -152,6 +152,7 @@ class AutoMixedPrecisionListsFp16 : public AutoMixedPrecisionLists {`
`152`	`152`	`list.insert("TmlpV3");`
`153`	`153`	`list.insert("Pmlp");`
`154`	`154`	`list.insert("FastUnsortedSegmentMax");`
	`155`	`+ list.insert("VoxelMax");`
`155`	`156`	`}`
`156`	`157`	`#if TENSORFLOW_USE_ROCM`
`157`	`158`	`if (true) {`