From b38c65c5801269c1111e9c1e94abbf153de39031 Mon Sep 17 00:00:00 2001 From: ojotoxy Date: Sun, 15 Mar 2020 12:02:59 -0700 Subject: [PATCH] Add support for FusedBatchNormV3 (#5065) No changes seem to be needed to _fused_batch_norm. It just works. --- python/tvm/relay/frontend/tensorflow.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python/tvm/relay/frontend/tensorflow.py b/python/tvm/relay/frontend/tensorflow.py index bed32b7274af..29d9d1bcb93b 100644 --- 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(),