Home
last modified time | relevance | path

Searched refs:buffers_to_assign (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dheap_simulator.h78 buffers_to_assign(nullptr) {} in Options()
87 const BufferValueFlatSet* buffers_to_assign; member
Dbuffer_assignment.cc1177 const flat_hash_set<const LogicalBuffer*>& buffers_to_assign = in AssignBuffersWithSequentialOrdering() local
1183 all_buffers_to_assign.insert(buffers_to_assign.begin(), in AssignBuffersWithSequentialOrdering()
1184 buffers_to_assign.end()); in AssignBuffersWithSequentialOrdering()
1195 options.buffers_to_assign = &buffer_value_set; in AssignBuffersWithSequentialOrdering()
1212 const flat_hash_set<const LogicalBuffer*>& buffers_to_assign = in AssignBuffersWithSequentialOrdering() local
1217 auto color_map = SplitBuffersByColor(buffers_to_assign); in AssignBuffersWithSequentialOrdering()
1225 options.buffers_to_assign = &buffer_value_set; in AssignBuffersWithSequentialOrdering()
Dheap_simulator.cc375 return options_.buffers_to_assign != nullptr && in IgnoreBuffer()
376 !options_.buffers_to_assign->contains(buffer); in IgnoreBuffer()
489 if (options_.buffers_to_assign != nullptr) { in Finish()
490 CHECK_EQ(options_.buffers_to_assign->size(), result.chunk_map.size()); in Finish()