[Benchmarking-Py] Deactivating TF_ENABLE_LAYOUT_NHWC=1

tensorflow · Sep 13, 2022 · a337598 · a337598
1 parent 7086a97
commit a337598
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tftrt/benchmarking-python/benchmark_runner.py b/tftrt/benchmarking-python/benchmark_runner.py
@@ -75,7 +75,7 @@ def __init__(self, args):
         self._args = args
 
         # Temporary fix to re-enable NHWC layout.
-        os.environ["TF_ENABLE_LAYOUT_NHWC"] = "1"
+        # os.environ["TF_ENABLE_LAYOUT_NHWC"] = "1"
 
         if args.use_xla_auto_jit:
             logging.info("[Benchmark] - Activating XLA JIT Auto Clustering")