From 29b27f3066ac99d3f01c65115bdd2b6cd4e2ccc7 Mon Sep 17 00:00:00 2001 From: Gaurav Jain Date: Sun, 10 Feb 2019 20:57:59 -0800 Subject: [PATCH] Fix tf.batch_gather rewrite rule tf.batch_gather cannot be simply changed to tf.gather without setting the batch_dims arg. PiperOrigin-RevId: 233338117 --- tensorflow/tools/compatibility/tf_upgrade_v2.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow/tools/compatibility/tf_upgrade_v2.py b/tensorflow/tools/compatibility/tf_upgrade_v2.py index 007f9b40ba4..4d87358829c 100644 --- a/tensorflow/tools/compatibility/tf_upgrade_v2.py +++ b/tensorflow/tools/compatibility/tf_upgrade_v2.py @@ -423,7 +423,7 @@ class TFAPIChangeSpec(ast_edits.APIChangeSpec): "tf.batch_to_space_nd": "tf.batch_to_space", "tf.batch_gather": - "tf.gather", + "tf.compat.v1.batch_gather", "tf.space_to_batch_nd": "tf.space_to_batch", "tf.nn.space_to_batch":