Searched refs:GetMutableAllocation (Results 1 – 2 of 2) sorted by relevance
537 BufferAllocation* GetMutableAllocation(BufferAllocation::Index index);
451 BufferAllocation* BufferAssignment::GetMutableAllocation( in GetMutableAllocation() function in xla::BufferAssignment1172 assignment->GetMutableAllocation(operand_slice.index()); in AssignSingleHloBuffer()1187 BufferAllocation* allocation = assignment->GetMutableAllocation( in AssignSingleHloBuffer()