diff --git a/python/tvm/relay/frontend/paddlepaddle.py b/python/tvm/relay/frontend/paddlepaddle.py index 1a6ceb843c0fb..d09e2c8e8da02 100644 --- a/python/tvm/relay/frontend/paddlepaddle.py +++ b/python/tvm/relay/frontend/paddlepaddle.py @@ -1937,8 +1937,9 @@ def convert_swish(g, op, block): """Operator converter for swish.""" x = g.get_node(op.input("X")[0]) - dtype = infer_type(x).checked_type.dtype - out = x / (_op.const(1.0, dtype) + _op.exp(_op.const(-1.0, dtype) * x)) + beta = op.attr("beta") + assert beta == 1.0, "Only support beta==1.0 for PaddlePaddle's swish" + out = x * _op.tensor.sigmoid(x) g.add_node(op.output("Out")[0], out)