Skip to content

Commit adb14fd

Browse files
codeislife99Tushar Dey
authored and
Tushar Dey
committed
Update tensorrt.py (#178)
1 parent 1a791ad commit adb14fd

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

python/tvm/relay/tensorrt.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -833,7 +833,8 @@ def is_valid_subgraph(func):
833833
# Scalar inputs not allowed
834834
if len(tupe_type.shape) == 0:
835835
return False
836-
input_batch_sizes.append(int(tupe_type.shape[0]))
836+
if not isinstance(tupe_type.shape[0], tvm.tir.expr.Any):
837+
input_batch_sizes.append(int(tupe_type.shape[0]))
837838
else:
838839
# Scalar inputs not allowed
839840
if len(var.checked_type.shape) == 0:

0 commit comments

Comments
 (0)