Searched refs:GetMutableAllocation (Results 1 – 2 of 2) sorted by relevance
436 BufferAllocation* BufferAssignment::GetMutableAllocation( in GetMutableAllocation() function in xla::BufferAssignment1228 assignment->GetMutableAllocation(operand_slice.index()); in AssignSingleHloBuffer()1243 BufferAllocation* allocation = assignment->GetMutableAllocation( in AssignSingleHloBuffer()
541 BufferAllocation* GetMutableAllocation(BufferAllocation::Index index);