Update _Input.py (#2293)

phyboy · web-flow · commit 6f0e265a34d6 · 2023-09-08T11:58:22.000-07:00
diff --git a/py/torch_tensorrt/_Input.py b/py/torch_tensorrt/_Input.py
@@ -265,7 +265,7 @@ def _parse_format(format: Any) -> _enums.TensorFormat:
                 return _enums.TensorFormat.channels_last
             else:
                 raise ValueError(
-                    "Provided an unsupported tensor format (support: NHCW/contiguous_format, NHWC/channel_last)"
+                    "Provided an unsupported tensor format (support: NCHW/contiguous_format, NHWC/channel_last)"
                 )
 
         elif isinstance(format, _enums.TensorFormat):

Original file line number	Diff line number	Diff line change
`@@ -265,7 +265,7 @@ def _parse_format(format: Any) -> _enums.TensorFormat:`
`265`	`265`	`return _enums.TensorFormat.channels_last`
`266`	`266`	`else:`
`267`	`267`	`raise ValueError(`
`268`		`- "Provided an unsupported tensor format (support: NHCW/contiguous_format, NHWC/channel_last)"`
	`268`	`+ "Provided an unsupported tensor format (support: NCHW/contiguous_format, NHWC/channel_last)"`
`269`	`269`	`)`
`270`	`270`
`271`	`271`	`elif isinstance(format, _enums.TensorFormat):`