[XLA] GTE of a certain element of the tuple does not need not keep other elements alive.
This achieves two things: 1. Heap simulation runtime is no longer quadratic in the number of tuple elements (as we don't add each GetTupleElement to the liveset of each buffer defined by the tuple). 2. A reduction in the heap memory footprint. PiperOrigin-RevId: 187079787
Loading
Please sign in to comment