Skip to content

T
tic
Unverified Commit b38c65c5 by ojotoxy Committed by GitHub
Options

Add support for FusedBatchNormV3 (#5065) 

No changes seem to be needed to _fused_batch_norm. It just works.
parent f7488ccf
Showing with 1 additions and 0 deletions
python/tvm/relay/frontend/tensorflow.py
...@@ -1561,6 +1561,7 @@ _convert_map = { ...@@ -1561,6 +1561,7 @@ _convert_map = {
'FloorMod' : _floormod(), 'FloorMod' : _floormod(),
'FusedBatchNorm' : _fused_batch_norm(), 'FusedBatchNorm' : _fused_batch_norm(),
'FusedBatchNormV2' : _fused_batch_norm(), 'FusedBatchNormV2' : _fused_batch_norm(),
'FusedBatchNormV3' : _fused_batch_norm(),
'Gather' : _gather(), 'Gather' : _gather(),
'GatherNd' : _gather_nd(), 'GatherNd' : _gather_nd(),
'GatherV2' : _gather(), 'GatherV2' : _gather(),
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment