This is an automated email from the ASF dual-hosted git repository. mbrookhart 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 1466e27 ignore 'training_mode' tag from onnx in batch_norm op (#9575) 1466e27 is described below commit 1466e27c45895119354b313aff7bff66616ce80a Author: Valery Chernov <black.che...@gmail.com> AuthorDate: Thu Nov 25 00:46:31 2021 +0300 ignore 'training_mode' tag from onnx in batch_norm op (#9575) Co-authored-by: Valery Chernov <valery.cher...@deelvin.com> --- python/tvm/relay/frontend/onnx.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/python/tvm/relay/frontend/onnx.py b/python/tvm/relay/frontend/onnx.py index d55c0e5..189c7f0 100644 --- a/python/tvm/relay/frontend/onnx.py +++ b/python/tvm/relay/frontend/onnx.py @@ -469,8 +469,10 @@ class BatchNorm(OnnxOpConverter): @classmethod def _impl_v1(cls, inputs, attr, params): # TODO(zhreshold): 'spatial' is not properly handled here. + # TODO(vvchernov): 'training_mode' (onnx tag) is not correctly handled, ignore for now out = AttrCvt( - op_name="batch_norm", ignores=["spatial", "is_test", "consumed_inputs", "momentum"] + op_name="batch_norm", + ignores=["spatial", "is_test", "consumed_inputs", "momentum", "training_mode"], )(inputs, attr, params) return out[0]