This is an automated email from the ASF dual-hosted git repository.
masahi pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-tvm.git
The following commit(s) were added to refs/heads/master by this push:
new 10c2b7f [MXNET]abs, round, reciprocal, sign, softsign, hard_sigmoid
(#5587)
10c2b7f is described below
commit 10c2b7ff58c64694c5d99ad7d85cee5bad19c4ed
Author: Samuel <siju.sam...@huawei.com>
AuthorDate: Thu May 14 02:31:36 2020 +0530
[MXNET]abs, round, reciprocal, sign, softsign, hard_sigmoid (#5587)
---
python/tvm/relay/frontend/mxnet.py | 19 +++++++++++++++++++
tests/python/frontend/mxnet/test_forward.py | 4 +++-
2 files changed, 22 insertions(+), 1 deletion(-)
diff --git a/python/tvm/relay/frontend/mxnet.py
b/python/tvm/relay/frontend/mxnet.py
index 4cb7a2a..4c3144c 100644
--- a/python/tvm/relay/frontend/mxnet.py
+++ b/python/tvm/relay/frontend/mxnet.py
@@ -789,6 +789,19 @@ def _mx_l2_normalize(inputs, attrs):
return _op.nn.l2_normalize(inputs[0], **new_attrs)
+def _mx_softsign(inputs, attrs):
+ return inputs[0] / (_expr.const(1.0) + _op.abs(inputs[0]))
+
+
+def _mx_hard_sigmoid(inputs, attrs):
+ x = (_expr.const(0.2) * inputs[0]) + _expr.const(0.5)
+ return _op.clip(x, a_min=0.0, a_max=1.0)
+
+
+def _mx_reciprocal(inputs, attrs):
+ return _expr.const(1.0) /inputs[0]
+
+
def _mx_shape_array(inputs, attrs):
assert len(inputs) == 1
if attrs.get_int("lhs_begin", None) is not None:
@@ -1742,12 +1755,15 @@ def _mx_broadcast_logical(logical_op):
# Note: due to attribute conversion constraint
# ops in the identity set must be attribute free
_identity_list = [
+ "abs",
"log",
"exp",
"erf",
"sqrt",
"floor",
"ceil",
+ "round",
+ "sign",
"sigmoid",
"negative",
"reshape_like",
@@ -1856,6 +1872,9 @@ _convert_map = {
"softmax" : _softmax_op(_op.nn.softmax),
"log_softmax" : _softmax_op(_op.nn.log_softmax),
"Softmax" : _softmax_op(_op.nn.softmax),
+ "softsign" : _mx_softsign,
+ "hard_sigmoid" : _mx_hard_sigmoid,
+ "reciprocal" : _mx_reciprocal,
# per op specialization
"Reshape" : _reshape,
"reshape" : _reshape,
diff --git a/tests/python/frontend/mxnet/test_forward.py
b/tests/python/frontend/mxnet/test_forward.py
index 3fb8e30..9dd8506 100644
--- a/tests/python/frontend/mxnet/test_forward.py
+++ b/tests/python/frontend/mxnet/test_forward.py
@@ -365,7 +365,9 @@ def test_forward_elemwise_ops():
def test_forward_unary_ops():
- for op in ["cos", "sin", "tan",
+ for op in ["abs", "sqrt", "ceil", "floor", "round", "reciprocal",
+ "softsign", "hard_sigmoid",
+ "cos", "sin", "tan",
"cosh", "sinh", "tanh",
"arccos", "arcsin", "arctan",
"arccosh", "arcsinh", "arctanh"]:
