remove unnecessary mask

trevor-m · May 29, 2021 · 8afbd30 · 8afbd30
1 parent ff870f7
commit 8afbd30
Showing 1 changed file with 0 additions and 1 deletion.
diff --git 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(