[Benchmarking Py] TF_ENABLE_LAYOUT_NHWC=1 added

Author: DEKHTIARJonathan

tftrt/benchmarking-python/benchmark_runner.py

@@ -74,6 +74,9 @@ def evaluate_model(self, predictions, expected, bypass_data_to_eval):
     def __init__(self, args):
         self._args = args
 
+        # Temporary fix to re-enable NHWC layout.
+        os.environ["TF_ENABLE_LAYOUT_NHWC"] = "1"
+
         if args.use_xla_auto_jit:
             logging.info("[Benchmark] - Activating XLA JIT Auto Clustering")
             os.environ["TF_XLA_FLAGS"] = "--tf_xla_auto_jit=2"