Home
last modified time | relevance | path

Searched refs:GlobalDecreasingSizeBestFitHeap (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dheap_simulator.cc507 GlobalDecreasingSizeBestFitHeap<BufferType>::GlobalDecreasingSizeBestFitHeap( in GlobalDecreasingSizeBestFitHeap() function in xla::GlobalDecreasingSizeBestFitHeap
519 typename GlobalDecreasingSizeBestFitHeap<BufferType>::BufferIntervalCompare
520 GlobalDecreasingSizeBestFitHeap<BufferType>::GetTemporalBufferIntervalCompare() in GetTemporalBufferIntervalCompare()
545 /*static*/ typename GlobalDecreasingSizeBestFitHeap<
547 GlobalDecreasingSizeBestFitHeap<BufferType>::GetSpatialBufferIntervalCompare() { in GetSpatialBufferIntervalCompare()
560 void GlobalDecreasingSizeBestFitHeap<BufferType>::Alloc( in Alloc()
575 void GlobalDecreasingSizeBestFitHeap<BufferType>::ShareWith( in ShareWith()
592 GlobalDecreasingSizeBestFitHeap<BufferType>::GetTransitiveColocations( in GetTransitiveColocations()
609 void GlobalDecreasingSizeBestFitHeap<BufferType>::Free(const BufferType* buffer, in Free()
818 GlobalDecreasingSizeBestFitHeap<BufferType>::Finish() { in Finish()
[all …]
Dmemory_space_assignment_best_fit_repacker.cc25 using Type = GlobalDecreasingSizeBestFitHeap<AllocationBlock>::Type;
31 : public GlobalDecreasingSizeBestFitHeap<AllocationBlock> {
34 : GlobalDecreasingSizeBestFitHeap<AllocationBlock>(alignment, type), in BestFitRepacker()
Dmemory_space_assignment_best_fit_repacker.h29 using Type = GlobalDecreasingSizeBestFitHeap<AllocationBlock>::Type;
33 Type type = GlobalDecreasingSizeBestFitHeap<AllocationBlock>::kTemporal)
Dheap_simulator.h358 class GlobalDecreasingSizeBestFitHeap : public HeapAlgorithm<BufferType> {
390 explicit GlobalDecreasingSizeBestFitHeap(int64 alignment,
392 ~GlobalDecreasingSizeBestFitHeap() override {} in ~GlobalDecreasingSizeBestFitHeap()
480 : public GlobalDecreasingSizeBestFitHeap<HloValue> {
484 : GlobalDecreasingSizeBestFitHeap<HloValue>(alignment, type),
Dmemory_space_assignment_utils.h29 const GlobalDecreasingSizeBestFitHeap<HloValue>::BufferInterval&
Dmemory_space_assignment.h109 const GlobalDecreasingSizeBestFitHeap<HloValue>::BufferInterval& interval,
241 const GlobalDecreasingSizeBestFitHeap<HloValue>::BufferInterval& interval) in BufferIntervalAlternateMemoryBenefit()
344 const GlobalDecreasingSizeBestFitHeap<HloValue>::BufferInterval& interval)
392 GlobalDecreasingSizeBestFitHeap<HloValue>::BufferInterval;
394 GlobalDecreasingSizeBestFitHeap<HloValue>::BufferIntervalCompare;
942 : public GlobalDecreasingSizeBestFitHeap<HloValue> {
952 : GlobalDecreasingSizeBestFitHeap(options.alignment_in_bytes), in AlternateMemoryBestFitHeap()
Dmemory_space_assignment_utils.cc110 const GlobalDecreasingSizeBestFitHeap<HloValue>::BufferInterval& interval) { in IsIntervalAllowedInAlternateMemory()
Dheap_simulator_test.cc1020 : public GlobalDecreasingSizeBestFitHeap<HloValue> {
1023 : GlobalDecreasingSizeBestFitHeap(/*alignment=*/1) {} in InheritedGlobalDecreasingSizeBestFitHeap()
1034 chunk_candidate_ = GlobalDecreasingSizeBestFitHeap::FindChunkCandidate( in FindChunkCandidate()
1042 GlobalDecreasingSizeBestFitHeap::CommitChunk(buffer_interval_, in CommitChunk()
1055 GlobalDecreasingSizeBestFitHeap<HloValue> heap(/*alignment=*/1); in TEST_F()
1076 GlobalDecreasingSizeBestFitHeap<HloValue> heap(/*alignment=*/1); in TEST_F()
1118 GlobalDecreasingSizeBestFitHeap<HloValue> heap(/*alignment=*/20); in TEST_F()
1162 GlobalDecreasingSizeBestFitHeap<HloValue> heap(/*alignment=*/1); in TEST_F()
1201 GlobalDecreasingSizeBestFitHeap<HloValue> heap(/*alignment=*/1); in TEST_F()
1232 GlobalDecreasingSizeBestFitHeap<HloValue> heap(/*alignment=*/1); in TEST_F()
[all …]
Dbuffer_assignment.cc1424 GlobalDecreasingSizeBestFitHeap<HloValue>::kSpatial)); in AssignBuffersWithSequentialOrdering()
1428 GlobalDecreasingSizeBestFitHeap<HloValue>::kTemporal)); in AssignBuffersWithSequentialOrdering()
Dmemory_space_assignment.cc186 const GlobalDecreasingSizeBestFitHeap<HloValue>::BufferInterval& interval, in GetMemoryBoundedness()
701 const GlobalDecreasingSizeBestFitHeap<HloValue>::BufferInterval& interval) in BufferIntervalAlternateMemoryBenefit()
2640 return GlobalDecreasingSizeBestFitHeap< in GetMemoryBoundednessBufferIntervalCompare()