Commit c6312773 authored by Michael Kuperstein's avatar Michael Kuperstein Committed by Gunhan Gulsoy
Browse files

[XLA] Do not recompute flattened sets inside layout assignment. 

Cache the flattened sets instead of recomputing them. This matters for large graphs, since we may request the flattened set thousands of times on the same instruction, and it may be fairly expensive to construct for large tuples.

PiperOrigin-RevId: 187046642
parent f487340e
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