Skip to content

Commit 2f91fb6

Browse files
committed
Update _Input.py
Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
1 parent ee07cff commit 2f91fb6

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

py/torch_tensorrt/_Input.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -125,7 +125,7 @@ def __str__(self) -> str:
125125
return "Input(shape={}, dtype={}, format={})".format(self.shape, str(self.dtype), str(self.format))
126126
elif self.shape_mode == Input._ShapeMode.DYNAMIC:
127127
return "Input(min_shape={}, opt_shape={}, max_shape={}, dtype={}, format={})".format(
128-
self.shape["min_shape"], self.shape["min_shape"], self.shape["min_shape"], str(self.dtype),
128+
self.shape["min_shape"], self.shape["opt_shape"], self.shape["max_shape"], str(self.dtype),
129129
str(self.format))
130130
else:
131131
raise RuntimeError("Unknown input shape mode")
@@ -145,7 +145,7 @@ def _to_internal(self) -> _C.Input:
145145
"Input shape specifications for inputs are required to be a List, tuple or torch.Size, found type: "
146146
+ str(type(self.shape["opt_shape"])) + " for opt_shape")
147147
else:
148-
internal_in.min = self.shape["op_shape"]
148+
internal_in.min = self.shape["opt_shape"]
149149

150150
if not Input._supported_input_size_type(self.shape["max_shape"]):
151151
raise TypeError(

0 commit comments

Comments
 (0)