Searched refs:gte_cache (Results 1 – 1 of 1) sorted by relevance
1563 absl::flat_hash_map<int64, HloInstruction*> gte_cache; in RematerializeInstructions() local1571 auto cached_gte = gte_cache.find(*user.index); in RematerializeInstructions()1572 if (cached_gte == gte_cache.end()) { in RematerializeInstructions()1579 gte_cache[*user.index] = remat_use; in RematerializeInstructions()