This is an automated email from the ASF dual-hosted git repository. andrewzhaoluo pushed a commit to branch aluo/potential-turing-vortex-changes in repository https://gitbox.apache.org/repos/asf/tvm.git
commit 23d90bd41aa3b9f71980ccbe674584bbd9fcd6b7 Author: Andrew Zhao Luo <[email protected]> AuthorDate: Mon Nov 7 09:39:23 2022 -0800 use builtin layernorm --- python/tvm/relay/frontend/onnx.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/tvm/relay/frontend/onnx.py b/python/tvm/relay/frontend/onnx.py index e244b4d9a1..393f7c56b6 100644 --- a/python/tvm/relay/frontend/onnx.py +++ b/python/tvm/relay/frontend/onnx.py @@ -1048,7 +1048,7 @@ class EmbedLayerNormalization(OnnxOpConverter): if segment_ids: vec_sum = _op.add(vec_sum, segment_vec) - ln = layer_norm(vec_sum, eps, gamma, beta) + ln = _op.nn.layer_norm(vec_sum, gamma, beta, epsilon=eps, axis=-1) mask_index = _op.const(np.zeros((batch_size,), dtype="int32")) if mask:
