projects
/
platform
/
upstream
/
tvm.git
/ commitdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
| commitdiff |
tree
raw
|
patch
| inline |
side by side
(parent:
2421a85
)
Support for AddV2 in Relay Tensorflow frontend converter. (#5046)
author
Fernand Pajot
<accounts@epigram.me>
Thu, 12 Mar 2020 07:09:00 +0000
(
01:09
-0600)
committer
GitHub
<noreply@github.com>
Thu, 12 Mar 2020 07:09:00 +0000
(
00:09
-0700)
python/tvm/relay/frontend/tensorflow.py
patch
|
blob
|
history
diff --git
a/python/tvm/relay/frontend/tensorflow.py
b/python/tvm/relay/frontend/tensorflow.py
index
f1cb815
..
bed32b7
100644
(file)
--- 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'),