Add support for FusedBatchNormV3 (apache#5065)

No changes seem to be needed to _fused_batch_norm. It just works.
neo-ai · Apr 17, 2020 · 9ff736f · 9ff736f
1 parent 1ebe9f0
commit 9ff736f
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(),