remove unnecessary mask

masahi · May 30, 2021 · 47bdef9 · 47bdef9
1 parent 445a7da
commit 47bdef9
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(