experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Bug fix for tf_upgrade_v2

Browse Source 
PiperOrigin-RevId: 242185991
This commit is contained in:
Edward Loper 2019-04-05 13:43:25 -07:00 committed by TensorFlower Gardener
parent 112e128f39
commit f8dfa51551
2 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/tools/compatibility/tf_upgrade_v2.py
View File

@ -1832,8 +1832,8 @@ def _add_argument_transformer(parent, node, full_name, name, logs,
node.keywords.append(ast.keyword(arg=arg_name, value=arg_value_ast))
logs.append((
ast_edits.INFO, node.lineno, node.col_offset,
"Adding argument '%s' to call to %s." % (pasta.dump(node.keywords[-1],
full_name or name))
"Adding argument '%s' to call to %s." % (pasta.dump(node.keywords[-1]),
full_name or name)
))
return node

6
tensorflow/tools/compatibility/tf_upgrade_v2_test.py
View File

@ -1566,6 +1566,12 @@ def _log_prob(self, x):
_, _, _, new_text = self._upgrade(text)
self.assertEqual(new_text, expected_text)
def testNnErosion2d(self):
text = "tf.nn.erosion2d(v, k, s, r, p)"
expected_text = "tf.nn.erosion2d(v, k, s, r, p, data_format='NHWC')"
_, _, _, new_text = self._upgrade(text)
self.assertEqual(new_text, expected_text)
class TestUpgradeFiles(test_util.TensorFlowTestCase):