Add return false

apache · Dec 3, 2020 · c03ece8 · c03ece8
1 parent 27982b8
commit c03ece8
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/python/tvm/relay/op/contrib/tensorrt.py b/python/tvm/relay/op/contrib/tensorrt.py
@@ -343,6 +343,7 @@ def batch_norm_annotate_fn(expr):  # pylint: disable=unused-variable
         return False
     if len(args[0].checked_type.shape) == 5 and get_tensorrt_version() < (6, 0, 1):
         logger.info("nn.batch_norm: TensorRT 6.0.1 or higher is required for rank 5 inputs.")
+        return False
     if len(args[0].checked_type.shape) > 5:
         logger.info("nn.batch_norm: Input rank must be 5 or less.")
         return False