Support for AddV2 in Relay Tensorflow frontend converter. (#5046)

95e7e34f · Fernand Pajot · GitHub · 2421a854 · 95e7e34f
Unverified Commit 95e7e34f authored Mar 12, 2020 by Fernand Pajot Committed by GitHub Mar 12, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 additions and 0 deletions

python/tvm/relay/frontend/tensorflow.py
+1 -0

No files found.
--- a/python/tvm/relay/frontend/tensorflow.py
+++ b/python/tvm/relay/frontend/tensorflow.py
@@ -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'),