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]
