No public description

PiperOrigin-RevId: 713096560

Author: tensorflower-gardener

official/benchmark/keras_imagenet_benchmark.py

@@ -448,6 +448,16 @@ def benchmark_1_gpu_fp16(self):
         dtype='float16',
         per_replica_batch_size=256)
 
+  def benchmark_1_gpu_fp32(self):
+    """Tests Keras model with 1 GPU and fp32."""
+    self._setup()
+    self._run_and_report_benchmark(
+        experiment_name='benchmark_1_gpu_fp32',
+        num_gpus=1,
+        distribution_strategy='one_device',
+        dtype='float32',
+        per_replica_batch_size=256)
+
   def benchmark_1_gpu_fp16_128_batch(self):
     """Tests Keras model with 1 GPU and fp16."""
     self._setup()