You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by ma...@apache.org on 2021/03/16 05:11:04 UTC
[tvm] branch main updated: init the concat tensor with 1s and then
slice them away (#7666)
This is an automated email from the ASF dual-hosted git repository.
masahi pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/main by this push:
new 348d4e7 init the concat tensor with 1s and then slice them away (#7666)
348d4e7 is described below
commit 348d4e7532f7d5ce68d76e7634a5f11b7dc60ab8
Author: Matthew Brookhart <mb...@octoml.ai>
AuthorDate: Mon Mar 15 23:10:44 2021 -0600
init the concat tensor with 1s and then slice them away (#7666)
---
python/tvm/relay/frontend/onnx.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/python/tvm/relay/frontend/onnx.py b/python/tvm/relay/frontend/onnx.py
index 4a0a1ed..391eaaa 100644
--- a/python/tvm/relay/frontend/onnx.py
+++ b/python/tvm/relay/frontend/onnx.py
@@ -2629,10 +2629,10 @@ class NonMaxSuppression(OnnxOpConverter):
# Call the second loop, rework outputs into correct form
init_count = _op.const(np.array([0]).astype("int64"), dtype="int64")
- init_out = _op.const(np.array([]).reshape([0, 3]).astype("int64"), dtype="int64")
+ init_out = _op.const(np.array([1, 1, 1]).reshape([1, 3]).astype("int64"), dtype="int64")
loop_vals = outer_loop(init_count, B, C, onnx_output, nms_size_output, init_out)
-
- return _expr.TupleGetItem(loop_vals, 5)
+ loop_out = _expr.TupleGetItem(loop_vals, 5)
+ return _op.strided_slice(loop_out, [1, 0], shape_of(loop_out), [1, 1])
# compatible operators that do NOT require any conversion.