Searched refs:GetAssignedAllocation (Results 1 – 3 of 3) sorted by relevance
402 const BufferAllocation& BufferAssignment::GetAssignedAllocation( in GetAssignedAllocation() function in xla::BufferAssignment408 const BufferAllocation& BufferAssignment::GetAssignedAllocation( in GetAssignedAllocation() function in xla::BufferAssignment410 return GetAssignedAllocation(*hlo_buffer.values()[0]); in GetAssignedAllocation()415 return const_cast<BufferAllocation*>(&GetAssignedAllocation(buffer)); in GetMutableAssignedAllocation()424 result.insert(GetAssignedAllocation(*value).GetSlice(*value)); in GetAllSlices()447 GetAssignedAllocation(*value); in GetInstructionAllocation()483 GetAssignedAllocation(*value).GetSlice(*value); in GetUniqueSlice()
386 const BufferAllocation& GetAssignedAllocation(const HloValue& value) const;388 const BufferAllocation& GetAssignedAllocation(
176 assignment.GetAssignedAllocation(*buffer).GetSlice(*buffer); in GetNoaliasMetadataForBuffer()