Add support for FusedBatchNormV3 (#5065)

No changes seem to be needed to _fused_batch_norm. It just works.
apache · Mar 15, 2020 · b38c65c · b38c65c
1 parent f7488cc
commit b38c65c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/python/tvm/relay/frontend/tensorflow.py b/python/tvm/relay/frontend/tensorflow.py
@@ -1561,6 +1561,7 @@ def _impl(inputs, attr, params):
     'FloorMod'                          : _floormod(),
     'FusedBatchNorm'                    : _fused_batch_norm(),
     'FusedBatchNormV2'                  : _fused_batch_norm(),
+    'FusedBatchNormV3'                  : _fused_batch_norm(),
     'Gather'                            : _gather(),
     'GatherNd'                          : _gather_nd(),
     'GatherV2'                          : _gather(),