Searched refs:AddAssignment (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | buffer_assignment.cc | 244 void BufferAllocation::AddAssignment(const HloValue& buffer, int64_t offset, in AddAssignment() function in xla::BufferAllocation
572 AddAssignment(allocation, buffer, /*offset=*/0, size); in NewAllocation()
577 void BufferAssignment::AddAssignment(BufferAllocation* allocation, in AddAssignment() function in xla::BufferAssignment
587 allocation->AddAssignment(*buffer_value, offset, size); in AddAssignment()
598 void BufferAssignment::AddAssignment(BufferAllocation* allocation, in AddAssignment() function in xla::BufferAssignment
601 allocation->AddAssignment(value, offset, size); in AddAssignment()
672 combined_allocation->AddAssignment(*value, base + offset, size); in CombineTempAllocations()
1106 assignment->AddAssignment(allocation, hlo_buffer, /*offset=*/0, in MaybeAssignBuffer()
1406 preset_allocations_iter->second->AddAssignment(*value, chunk.offset, in AssignPresetBuffers()
1627 assignment->AddAssignment(allocation, value, chunk.offset, chunk.size); in AssignBuffersFromHeapSimulator()
|
| D | buffer_assignment.h | 295 void AddAssignment(const HloValue& buffer, int64_t offset, int64_t size);
529 void AddAssignment(BufferAllocation* allocation, const HloBuffer& buffer,
532 void AddAssignment(BufferAllocation* allocation, const HloValue& value,
|