check batch_size is int

apache · Jun 2, 2021 · 8fa99ff · 8fa99ff
1 parent 8c92760
commit 8fa99ff
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/python/tvm/relay/frontend/tensorflow.py b/python/tvm/relay/frontend/tensorflow.py
@@ -817,6 +817,7 @@ def convert_combined_nms_with_all_class_nms(
     box_range = _op.arange(
         _op.const(0, dtype="int64"), _op.const(max_total_size, dtype="int64"), dtype="int64"
     )
+    assert isinstance(batch_size, int), "dynamic batch size not supported yet."
     tile_batch_reps = _op.const([batch_size, 1])
     box_range_2d = _op.tile(box_range, tile_batch_reps)
     valid_mask = _op.cast(