diff --git a/python/tvm/relay/frontend/tensorflow.py b/python/tvm/relay/frontend/tensorflow.py index 9a3d59e43dbc..a1c9d7efd4b5 100644 --- a/python/tvm/relay/frontend/tensorflow.py +++ b/python/tvm/relay/frontend/tensorflow.py @@ -841,7 +841,6 @@ def all_class_impl( _op.less(box_range_2d, _op.expand_dims(num_detections, axis=1)), "float32" ) nmsed_scores = nmsed_scores * valid_mask - nmsed_classes = nmsed_classes * valid_mask nmsed_boxes = nmsed_boxes * _op.expand_dims(valid_mask, axis=2) return _expr.TupleWrapper(