Searched defs:colocation (Results 1 – 3 of 3) sorted by relevance
524 for (auto colocation : GetTransitiveColocations(x)) { in GetTemporalBufferIntervalCompare() local529 for (auto colocation : GetTransitiveColocations(y)) { in GetTemporalBufferIntervalCompare() local882 for (auto colocation : GetTransitiveColocations(buffer_interval)) { in FindChunkCandidate() local962 for (auto colocation : GetTransitiveColocations(buffer_interval)) { in CommitChunk() local
1563 for (const HloValue* colocation : prefetch_candidate->colocations) { in AllocateCrossProgramPrefetchBuffer() local1901 for (auto& colocation : colocation_map) { in FinalizeAllocations() local
4393 for (const AllocationBlock* colocation : block->colocations) { in Repack() local