Searched refs:BufferAssigner (Results 1 – 6 of 6) sorted by relevance
274 friend class BufferAssigner;471 friend class BufferAssigner;536 class BufferAssigner {558 BufferAssigner(bool allocate_buffers_for_constants, in BufferAssigner() function564 virtual ~BufferAssigner() = default;660 TF_DISALLOW_COPY_AND_ASSIGN(BufferAssigner);
735 StatusOr<std::unique_ptr<BufferAssignment>> BufferAssigner::Run( in Run()741 BufferAssigner assigner(allocate_buffers_for_constants, std::move(colorer), in Run()824 bool BufferAssigner::MaybeAssignBuffer(BufferAllocation* allocation, in MaybeAssignBuffer()896 Status BufferAssigner::AssignBuffersForComputation( in AssignBuffersForComputation()1135 BufferAssigner::SplitBuffersByColor( in SplitBuffersByColor()1146 Status BufferAssigner::AssignBuffersWithSequentialOrdering( in AssignBuffersWithSequentialOrdering()1322 void BufferAssigner::AssignBuffersFromHeapSimulator( in AssignBuffersFromHeapSimulator()1362 void BufferAssigner::AddSetToColocatedBufferSets( in AddSetToColocatedBufferSets()1416 std::vector<BufferAssigner::ColocatedBufferSet>1417 BufferAssigner::MergeColocatedBufferSets( in MergeColocatedBufferSets()[all …]
91 return BufferAssigner::Run( in RunBufferAssignment()102 return BufferAssigner::Run( in RunBufferAssignmentNoBuffersForConstants()118 return BufferAssigner::Run( in RunBufferAssignmentNoBuffersReuseForAdd()131 return BufferAssigner::Run( in RunColoredBufferAssignment()145 return BufferAssigner::Run( in RunBufferAssignmentWithInstructionSequence()1892 return BufferAssigner::Run( in RunBufferAssignment()2212 BufferAssigner::Run( in TEST_F()2446 BufferAssigner::Run( in TEST_F()
63 auto status_or_buffer_assn = BufferAssigner::Run( in TEST_F()
645 BufferAssigner::Run(module.get(), in RunBackend()830 BufferAssigner::Run(module, in CompileAheadOfTime()
668 BufferAssigner::Run( in RunBackend()