Skip to content
This repository was archived by the owner on Feb 3, 2025. It is now read-only.

Commit a337598

Browse files
[Benchmarking-Py] Deactivating TF_ENABLE_LAYOUT_NHWC=1
1 parent 7086a97 commit a337598

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tftrt/benchmarking-python/benchmark_runner.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,7 @@ def __init__(self, args):
7575
self._args = args
7676

7777
# Temporary fix to re-enable NHWC layout.
78-
os.environ["TF_ENABLE_LAYOUT_NHWC"] = "1"
78+
# os.environ["TF_ENABLE_LAYOUT_NHWC"] = "1"
7979

8080
if args.use_xla_auto_jit:
8181
logging.info("[Benchmark] - Activating XLA JIT Auto Clustering")

0 commit comments

Comments
 (0)