Searched refs:AddAssignment (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | buffer_assignment.cc | 239 void BufferAllocation::AddAssignment(const HloValue& buffer, int64 offset, in AddAssignment() function in xla::BufferAllocation
567 AddAssignment(allocation, buffer, /*offset=*/0, size); in NewAllocation()
572 void BufferAssignment::AddAssignment(BufferAllocation* allocation, in AddAssignment() function in xla::BufferAssignment
582 allocation->AddAssignment(*buffer_value, offset, size); in AddAssignment()
593 void BufferAssignment::AddAssignment(BufferAllocation* allocation, in AddAssignment() function in xla::BufferAssignment
596 allocation->AddAssignment(value, offset, size); in AddAssignment()
667 combined_allocation->AddAssignment(*value, base + offset, size); in CombineTempAllocations()
1093 assignment->AddAssignment(allocation, hlo_buffer, /*offset=*/0, in MaybeAssignBuffer()
1393 preset_allocations_iter->second->AddAssignment(*value, chunk.offset, in AssignPresetBuffers()
1614 assignment->AddAssignment(allocation, value, chunk.offset, chunk.size); in AssignBuffersFromHeapSimulator()
|
| D | buffer_assignment.h | 292 void AddAssignment(const HloValue& buffer, int64 offset, int64 size);
526 void AddAssignment(BufferAllocation* allocation, const HloBuffer& buffer,
529 void AddAssignment(BufferAllocation* allocation, const HloValue& value,
|