Searched refs:RoundedBytes (Results 1 – 3 of 3) sorted by relevance
554 EXPECT_EQ(GPUBFCAllocator::RoundedBytes(size_t{2 << 20}), in TestForceAllowGrowth()562 EXPECT_EQ(GPUBFCAllocator::RoundedBytes(size_t{2 << 20}), in TestForceAllowGrowth()571 EXPECT_EQ(GPUBFCAllocator::RoundedBytes(size_t{2 << 20}), in TestForceAllowGrowth()580 EXPECT_EQ(GPUBFCAllocator::RoundedBytes(1LL << 31), in TestForceAllowGrowth()
53 RoundedBytes(std::min(total_memory, size_t{2 << 20})); in BFCAllocator()55 curr_region_allocation_bytes_ = RoundedBytes(total_memory); in BFCAllocator()138 bytes = RoundedBytes(bytes * kBackpedalFactor); in Extend()289 size_t BFCAllocator::RoundedBytes(size_t bytes) { in RoundedBytes() function in tensorflow::BFCAllocator399 size_t rounded_bytes = RoundedBytes(num_bytes); in AllocateRawInternal()
436 static size_t RoundedBytes(size_t bytes);