Support for AddV2 in Relay Tensorflow frontend converter. (#5046)
authorFernand Pajot <accounts@epigram.me>
Thu, 12 Mar 2020 07:09:00 +0000 (01:09 -0600)
committerGitHub <noreply@github.com>
Thu, 12 Mar 2020 07:09:00 +0000 (00:09 -0700)
python/tvm/relay/frontend/tensorflow.py

index f1cb815..bed32b7 100644 (file)
@@ -1522,6 +1522,7 @@ _freezed_graph_pruned_op_list = ['ReadVariableOp', 'ResourceGather', 'Variable',
 _convert_map = {
     'Abs'                               : AttrCvt('abs'),
     'Add'                               : _elemwise('add'),
+    'AddV2'                             : _elemwise('add'),
     'AddN'                              : _add_n(),
     'All'                               : _reduce('all'),
     'Any'                               : _reduce('any'),