fixed score gathering

masahi · May 29, 2021 · 115a5df · 115a5df
1 parent d203562
commit 115a5df
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tvm/relay/frontend/tensorflow.py b/python/tvm/relay/frontend/tensorflow.py
@@ -848,7 +848,7 @@ def _impl(inputs, attr, params, mod):
         indices_shape = _infer_shape(indices, mod)
         indices_dims = len(indices_shape)
         indices = _op.transpose(indices, axes=[-1] + list(range(indices_dims - 1)))
-        nmsed_scores = _op.gather_nd(scores, indices, batch_dims=1)
+        nmsed_scores = _op.gather_nd(scores_trans, indices, batch_dims=1)
 
         if attr["clip_boxes"]:
             nmsed_boxes = _op.maximum(nmsed_boxes, _expr.const(0, dtype="float32"))