Searched refs:prefetch_candidate (Results 1 – 2 of 2) sorted by relevance
1089 prefetch_candidate = FindCrossProgramPrefetchCandidate( in Finish() local1091 if (prefetch_candidate) { in Finish()1093 prefetch_candidate->buffer->instruction()->GetModule(); in Finish()1094 AllocateCrossProgramPrefetchBuffer(module, prefetch_candidate); in Finish()1527 HloModule* module, absl::optional<BufferInterval> prefetch_candidate) { in AllocateCrossProgramPrefetchBuffer() argument1528 if (!prefetch_candidate) { in AllocateCrossProgramPrefetchBuffer()1532 ChunkCandidate chunk_candidate = FindChunkCandidate(*prefetch_candidate); in AllocateCrossProgramPrefetchBuffer()1539 AddToPendingChunks(*prefetch_candidate, chunk_candidate); in AllocateCrossProgramPrefetchBuffer()1541 const HloValue* buffer = prefetch_candidate->buffer; in AllocateCrossProgramPrefetchBuffer()1548 prefetch_candidate->start, prefetch_candidate->end)); in AllocateCrossProgramPrefetchBuffer()[all …]
967 HloModule* module, absl::optional<BufferInterval> prefetch_candidate);