From 7e698397318a018022dc3127e2f68faedc250800 Mon Sep 17 00:00:00 2001 From: Matthew Barrett Date: Mon, 3 Feb 2020 16:14:11 +0000 Subject: [PATCH] Assert for incorrect number of inputs Change-Id: I393f3b3b62be73e427498d98456fb1d5a214e0af --- python/tvm/relay/frontend/tflite.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python/tvm/relay/frontend/tflite.py b/python/tvm/relay/frontend/tflite.py index ce2aaa6e947c..fbfa1fcd1563 100644 --- 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])