Commit 1fc63acf authored by Saurabh Saxena's avatar Saurabh Saxena Committed by TensorFlower Gardener
Browse files

Remove the resize op from the gradient of TensorListGather to avoid...

Remove the resize op from the gradient of TensorListGather to avoid implementing TensorListResize in XLA. Instead we directly pass a num_elements arg to TensorListScatter.

PiperOrigin-RevId: 231244477
parent 3b2be16d
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment