Assert for incorrect number of inputs

Change-Id: I393f3b3b62be73e427498d98456fb1d5a214e0af
apache · Feb 10, 2020 · 7e69839 · 7e69839
1 parent 7fca345
commit 7e69839
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/python/tvm/relay/frontend/tflite.py b/python/tvm/relay/frontend/tflite.py
@@ -1837,6 +1837,7 @@ def convert_detection_postprocess(self, op):
             )
 
         inputs = self.get_input_tensors(op)
+        assert len(inputs) == 3, "inputs length should be 3"
         cls_pred = self.get_expr(inputs[1].tensor_idx)
         loc_prob = self.get_expr(inputs[0].tensor_idx)
         anchor_values = self.get_tensor_value(inputs[2])