We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent f9acc9a commit d5e3a77Copy full SHA for d5e3a77
py/torch_tensorrt/dynamo/utils.py
@@ -347,7 +347,7 @@ def unwrap_tensor_shape(
347
if isinstance(tensor, int):
348
tensor_shape.append(tensor)
349
elif isinstance(tensor, torch.SymInt):
350
- min_max_opt: Dict[str, int] = extract_var_range_info(tensor)
+ min_max_opt = extract_var_range_info(tensor)
351
tensor_shape.append((min_max_opt["min"], min_max_opt["max"]))
352
elif isinstance(tensor, (torch.Tensor, FakeTensor)):
353
for dimension in tensor.shape:
0 commit comments