Searched refs:BufferAssigner (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | buffer_assignment.h | 288 friend class BufferAssigner;
495 friend class BufferAssigner;
585 class BufferAssigner {
623 BufferAssigner(bool allocate_buffers_for_constants, Colorer colorer, in BufferAssigner() function
630 virtual ~BufferAssigner() = default;
710 TF_DISALLOW_COPY_AND_ASSIGN(BufferAssigner);
|
| D | buffer_assignment.cc | 909 StatusOr<std::unique_ptr<BufferAssignment>> BufferAssigner::Run( in Run()
913 bool allocate_buffers_for_constants, BufferAssigner::Colorer colorer, in Run()
917 BufferAssigner assigner(allocate_buffers_for_constants, std::move(colorer), in Run()
924 bool BufferAssigner::LiveRangeInterferes(const HloValue* buffer1, in LiveRangeInterferes()
984 bool BufferAssigner::MaybeAssignBuffer(BufferAllocation* allocation, in MaybeAssignBuffer()
1098 Status BufferAssigner::AssignSingleHloBuffer( in AssignSingleHloBuffer()
1237 Status BufferAssigner::AssignBuffersForComputations( in AssignBuffersForComputations()
1347 BufferAssigner::SplitBuffersByColor( in SplitBuffersByColor()
1356 Status BufferAssigner::AssignPresetBuffers( in AssignPresetBuffers()
1407 Status BufferAssigner::AssignBuffersWithSequentialOrdering( in AssignBuffersWithSequentialOrdering()
[all …]
|
| D | buffer_assignment_test.cc | 92 return BufferAssigner::Run( in RunBufferAssignment()
102 return BufferAssigner::Run( in RunBufferAssignmentNoBuffersForConstants()
114 return BufferAssigner::Run( in RunBufferAssignmentNoBuffersReuseForAdd()
119 /*colorer=*/BufferAssigner::DefaultColorer(), in RunBufferAssignmentNoBuffersReuseForAdd()
125 HloModule* module, BufferAssigner::Colorer colorer, int64 alignment = 1) { in RunColoredBufferAssignment()
126 return BufferAssigner::Run( in RunColoredBufferAssignment()
139 return BufferAssigner::Run( in RunBufferAssignmentWithInstructionSequence()
150 return BufferAssigner::Run( in RunBufferAssignmentWithPresetAssignments()
155 BufferAssigner::DefaultColorer(), in RunBufferAssignmentWithPresetAssignments()
2041 return BufferAssigner::Run( in RunBufferAssignment()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| D | cpu_noalias_test.cc | 65 auto status_or_buffer_assn = BufferAssigner::Run( in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_compiler.cc | 534 BufferAssigner::Run( in RunHloPassesAndBufferAssignement()
540 /*colorer=*/BufferAssigner::DefaultColorer(), in RunHloPassesAndBufferAssignement()
577 BufferAssigner::Run( in CompileModuleToLlvmIrImpl()
583 /*colorer=*/BufferAssigner::DefaultColorer(), in CompileModuleToLlvmIrImpl()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_compiler.cc | 609 BufferAssigner::Run(module.get(), in RunHloPassesAndBufferAssignement()
725 BufferAssigner::Run(module.get(), in RunBackend()
921 BufferAssigner::Run(module, in CompileAheadOfTime()
|