Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dmemory_space_assignment.cc1220 RequiredMemoryAssignmentAt(while_value, while_time); in IsUseAllowedInAlternateMemory()
2308 AlternateMemoryBestFitHeap::RequiredMemoryAssignmentAt(const HloValue* buffer, in RequiredMemoryAssignmentAt() function in xla::memory_space_assignment::AlternateMemoryBestFitHeap
2336 RequiredMemoryAssignmentAt(value, time); in AliasedRequiredAssignmentForUse()
2365 auto existing_required_assignment = RequiredMemoryAssignmentAt(value, time); in AddRequiredAssignment()
2740 auto required_assignment_at_start = RequiredMemoryAssignmentAt( in AllocateSegment()
2748 auto required_assignment_at_end = RequiredMemoryAssignmentAt( in AllocateSegment()
H A Dmemory_space_assignment.h1467 std::optional<RequiredMemoryAssignment> RequiredMemoryAssignmentAt(