Searched refs:RequiredMemoryAssignmentAt (Results 1 – 2 of 2) sorted by relevance
1220 RequiredMemoryAssignmentAt(while_value, while_time); in IsUseAllowedInAlternateMemory()2308 AlternateMemoryBestFitHeap::RequiredMemoryAssignmentAt(const HloValue* buffer, in RequiredMemoryAssignmentAt() function in xla::memory_space_assignment::AlternateMemoryBestFitHeap2336 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()
1467 std::optional<RequiredMemoryAssignment> RequiredMemoryAssignmentAt(