Assert for incorrect number of inputs

Change-Id: I393f3b3b62be73e427498d98456fb1d5a214e0af
apache · Feb 3, 2020 · c211f6a · c211f6a
1 parent 07dd73d
commit c211f6a
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
@@ -1693,6 +1693,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])