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

Fix tf.batch_gather rewrite rule

Browse Source 
tf.batch_gather cannot be simply changed to tf.gather without setting
the batch_dims arg.

PiperOrigin-RevId: 233338117
This commit is contained in:
Gaurav Jain 2019-02-10 20:57:59 -08:00 committed by TensorFlower Gardener
parent f501504583
commit 29b27f3066
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -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":