From 288888ee4d278539059a7d010a03b39ad7b539c5 Mon Sep 17 00:00:00 2001 From: ojotoxy Date: Fri, 13 Mar 2020 12:09:48 -0700 Subject: [PATCH] Add support for FusedBatchNormV3 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(),