This is an automated email from the ASF dual-hosted git repository.
zhic 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 40450af Support mxnet _contrib_SyncBatchNorm (#6245)
40450af is described below
commit 40450af983ab74385630620210ee4c670a0963db
Author: Trevor Morris <[email protected]>
AuthorDate: Wed Aug 12 09:00:55 2020 -0700
Support mxnet _contrib_SyncBatchNorm (#6245)
---
python/tvm/relay/frontend/mxnet.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/python/tvm/relay/frontend/mxnet.py
b/python/tvm/relay/frontend/mxnet.py
index 9a353f3..36ffe54 100644
--- a/python/tvm/relay/frontend/mxnet.py
+++ b/python/tvm/relay/frontend/mxnet.py
@@ -2312,6 +2312,7 @@ _convert_map = {
"Dropout" : _mx_dropout,
"BatchNorm" : _mx_batch_norm,
"BatchNorm_v1" : _mx_batch_norm,
+ "_contrib_SyncBatchNorm" : _mx_batch_norm,
"InstanceNorm" : _mx_instance_norm,
"LayerNorm" : _mx_layer_norm,
"LRN" : _mx_lrn,
