Home
last modified time | relevance | path

Searched refs:preferred_offset (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dheap_simulator.cc856 int64 preferred_offset) const { in FindChunkCandidate()
900 int64 preferred_chunk_end = preferred_offset + buffer_interval.size; in FindChunkCandidate()
907 if (free_offset <= preferred_offset && in FindChunkCandidate()
909 min_fit_chunk = {preferred_offset, buffer_interval.size}; in FindChunkCandidate()
911 free_offset <= preferred_offset) { in FindChunkCandidate()
914 min_fit_chunk = {preferred_offset, buffer_interval.size}; in FindChunkCandidate()
920 if ((preferred_offset < 0 || min_fit_chunk.offset != preferred_offset) && in FindChunkCandidate()
936 if (result_.heap_size <= preferred_offset) { in FindChunkCandidate()
938 min_fit_chunk = {preferred_offset, buffer_interval.size}; in FindChunkCandidate()
Dmemory_space_assignment.cc1302 AliasedOffset* preferred_offset = nullptr; in AllocateAllocationValues() local
1306 preferred_offset = preferred_offset_it->second; in AllocateAllocationValues()
1419 request.preferred_offset = preferred_offset; in AllocateAllocationValues()
2215 AliasedOffset* preferred_offset = nullptr; in AllocateInAlternateMemoryNoCopy() local
2217 preferred_offset = GetAliasedOffset(*prev_allocation); in AllocateInAlternateMemoryNoCopy()
2223 if (request.preferred_offset) { in AllocateInAlternateMemoryNoCopy()
2226 CHECK(!preferred_offset || request.preferred_offset == preferred_offset) in AllocateInAlternateMemoryNoCopy()
2227 << "preferred_offset = " << preferred_offset->offset in AllocateInAlternateMemoryNoCopy()
2228 << ", request.preferred_offset = " << request.preferred_offset->offset; in AllocateInAlternateMemoryNoCopy()
2229 preferred_offset = request.preferred_offset; in AllocateInAlternateMemoryNoCopy()
[all …]
Dheap_simulator.h426 int64 preferred_offset = -1) const;
Dmemory_space_assignment.h1045 AliasedOffset* preferred_offset; member
1196 const AllocationRequest& request, const AliasedOffset* preferred_offset,
Dheap_simulator_test.cc1029 int64 preferred_offset = -1) { in FindChunkCandidate() argument
1035 buffer_interval_, preferred_offset); in FindChunkCandidate()