Searched refs:RequiredMemoryAssignmentAt (Results 1 – 2 of 2) sorted by relevance
937 RequiredMemoryAssignmentAt(while_value, while_time); in IsUseAllowedInAlternateMemory()1634 AlternateMemoryBestFitHeap::RequiredMemoryAssignmentAt(const HloValue* buffer, in RequiredMemoryAssignmentAt() function in xla::AlternateMemoryBestFitHeap1662 RequiredMemoryAssignmentAt(value, time); in AliasedRequiredAssignmentForUse()1691 auto existing_required_assignment = RequiredMemoryAssignmentAt(value, time); in AddRequiredAssignment()1970 auto required_assignment_at_start = RequiredMemoryAssignmentAt( in AllocateSegment()1978 auto required_assignment_at_end = RequiredMemoryAssignmentAt( in AllocateSegment()
1200 absl::optional<RequiredMemoryAssignment> RequiredMemoryAssignmentAt(