This is an automated email from the ASF dual-hosted git repository.
kevinthesun pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-tvm.git
The following commit(s) were added to refs/heads/master by this push:
new b38c65c Add support for FusedBatchNormV3 (#5065)
b38c65c is described below
commit b38c65c5801269c1111e9c1e94abbf153de39031
Author: ojotoxy <[email protected]>
AuthorDate: Sun Mar 15 12:02:59 2020 -0700
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 bed32b7..29d9d1b 100644
--- a/python/tvm/relay/frontend/tensorflow.py
+++ b/python/tvm/relay/frontend/tensorflow.py
@@ -1561,6 +1561,7 @@ _convert_map = {
'FloorMod' : _floormod(),
'FusedBatchNorm' : _fused_batch_norm(),
'FusedBatchNormV2' : _fused_batch_norm(),
+ 'FusedBatchNormV3' : _fused_batch_norm(),
'Gather' : _gather(),
'GatherNd' : _gather_nd(),
'GatherV2' : _gather(),
