Home
last modified time | relevance | path

Searched refs:prefetch_candidate (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dmemory_space_assignment.cc1089 prefetch_candidate = FindCrossProgramPrefetchCandidate( in Finish() local
1091 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() argument
1528 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 …]
Dmemory_space_assignment.h967 HloModule* module, absl::optional<BufferInterval> prefetch_candidate);