Commit 29b27f30 authored by Gaurav Jain's avatar Gaurav Jain Committed by TensorFlower Gardener
Browse files

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
parent f5015045
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment