Support mxnet _contrib_SyncBatchNorm (apache#6245)

trevor-m · Sep 2, 2020 · 8fd5969 · 8fd5969
1 parent d0dc58d
commit 8fd5969
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/python/tvm/relay/frontend/mxnet.py b/python/tvm/relay/frontend/mxnet.py
@@ -2312,6 +2312,7 @@ def impl(inputs, input_types):
     "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,