From 47bdef9e0fcdbab4671dd46044be5acac24b2f2b Mon Sep 17 00:00:00 2001 From: Masahiro Masuda Date: Sat, 29 May 2021 19:47:04 +0900 Subject: [PATCH] remove unnecessary mask --- python/tvm/relay/frontend/tensorflow.py | 1 - 1 file changed, 1 deletion(-) 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(