Reuse the input buffers when possible in TensorList kernels.
This lets the tf runtime reuse the memory for the input tensor to the output tensor, preventing allocations and copying, and making the normal runtime of successive push-backs O(n) instead of O(n^2). PiperOrigin-RevId: 227056083
Loading
Please sign in to comment