Searched defs:colocation (Results 1 – 3 of 3) sorted by relevance
527 for (auto colocation : GetTransitiveColocations(x)) { in GetTemporalBufferIntervalCompare() local878 for (const BufferType* colocation : in FindChunkCandidate() local921 for (const BufferType* colocation : in FindChunkCandidate() local960 for (auto colocation : GetTransitiveColocations(buffer_interval)) { in CommitChunk() local
2139 for (const HloValue* colocation : prefetch_candidate->colocations) { in AllocateCrossProgramPrefetchBuffer() local2641 for (auto& colocation : colocation_map) { in FinalizeAllocations() local
5278 for (const AllocationBlock* colocation : block->colocations) { in Repack() local