Searched refs:GetMutableAllocation (Results 1 – 2 of 2) sorted by relevance
503 BufferAllocation* GetMutableAllocation(BufferAllocation::Index index);
389 BufferAllocation* BufferAssignment::GetMutableAllocation( in GetMutableAllocation() function in xla::BufferAssignment1055 assignment->GetMutableAllocation(operand_slice.index()); in AssignBuffersForComputation()1082 BufferAllocation* allocation = assignment->GetMutableAllocation( in AssignBuffersForComputation()