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 075e2ec7bb [Frontend][Paddle]fix eye and dist (#14292)
075e2ec7bb is described below
commit 075e2ec7bb9a09d37106e6bd93b48bf3bb0c3324
Author: xg <[email protected]>
AuthorDate: Wed Mar 15 12:52:58 2023 +0800
[Frontend][Paddle]fix eye and dist (#14292)
fix eye and dist
---
python/tvm/relay/frontend/paddlepaddle.py | 8 +++++---
1 file changed, 5 insertions(+), 3 deletions(-)
diff --git a/python/tvm/relay/frontend/paddlepaddle.py
b/python/tvm/relay/frontend/paddlepaddle.py
index f771e605ca..9b141fec6a 100755
--- a/python/tvm/relay/frontend/paddlepaddle.py
+++ b/python/tvm/relay/frontend/paddlepaddle.py
@@ -409,11 +409,11 @@ def convert_dist(g, op, block):
dtype = infer_type(x).checked_type.dtype
p = op.attr("p")
if p == np.inf:
- out = _op.reduce.max(_op.abs(z))
+ out = _op.reduce.max(z)
elif p == np.NINF:
- out = _op.reduce.min(_op.abs(z))
+ out = _op.reduce.min(z)
elif p == 0.0:
- out = _op.reduce.sum(_op.sign(_op.abs(z)))
+ out = _op.reduce.sum(_op.sign(z))
else:
inv_p = _expr.const(1.0 / p, dtype=dtype)
p = _expr.const(p, dtype=dtype)
@@ -576,6 +576,8 @@ def convert_eye(g, op, block):
num_rows = op.attr("num_rows")
num_columns = op.attr("num_columns")
+ if num_columns == -1:
+ num_columns = num_rows
one_nums = min(num_rows, num_columns)
dtype = op.attr("dtype")
dtype = _convert_dtype_value(dtype)
