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
Loading
Please sign in to comment