Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/
Dbfc_allocator.cc71 new (BinFromIndex(b)) Bin(this, bin_size); in BFCAllocator()
72 CHECK_EQ(BinForSize(bin_size), BinFromIndex(b)); in BFCAllocator()
73 CHECK_EQ(BinForSize(bin_size + 255), BinFromIndex(b)); in BFCAllocator()
74 CHECK_EQ(BinForSize(bin_size * 2 - 1), BinFromIndex(b)); in BFCAllocator()
76 CHECK_NE(BinForSize(bin_size * 2), BinFromIndex(b)); in BFCAllocator()
90 BinFromIndex(b)->~Bin(); in ~BFCAllocator()
471 if (!BinFromIndex(i)->free_chunks.empty()) { in LargestFreeChunk()
472 return ChunkFromHandle(*BinFromIndex(i)->free_chunks.rbegin())->size; in LargestFreeChunk()
529 Bin* b = BinFromIndex(bin_num); in FindChunkPtr()
723 Bin* new_bin = BinFromIndex(bin_num); in InsertFreeChunkIntoBin()
[all …]
Dbfc_allocator.h548 Bin* BinFromIndex(BinNum index) { in BinFromIndex() function
559 Bin* BinForSize(size_t bytes) { return BinFromIndex(BinNumForSize(bytes)); } in BinForSize()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_bfc_allocator_test.cc698 BFCAllocator::Bin* bin = a.BinFromIndex(i); in TestRegionDeallocation()