Searched refs:NewAllocation (Results 1 – 2 of 2) sorted by relevance
502 BufferAllocation* BufferAssignment::NewAllocation(const LogicalBuffer& buffer, in NewAllocation() function in xla::BufferAssignment998 assignment->NewAllocation(*buffer, buffer_size); in AssignBuffersForComputation()1015 assignment->NewAllocation(*buffer, buffer_size); in AssignBuffersForComputation()1029 assignment->NewAllocation(*buffer, buffer_size); in AssignBuffersForComputation()1038 assignment->NewAllocation(*buffer, buffer_size); in AssignBuffersForComputation()1123 assignment->NewAllocation(*buffer, buffer_size); in AssignBuffersForComputation()1727 allocation = assignment->NewAllocation(*buffer, buffer_size); in AssignColocatedBufferSets()
488 BufferAllocation* NewAllocation(const LogicalBuffer& buffer, int64 size);