Searched refs:RoundedBytes (Results 1 – 3 of 3) sorted by relevance
43 RoundedBytes(std::min(total_memory, size_t{1048576})); in BFCAllocator()45 curr_region_allocation_bytes_ = RoundedBytes(total_memory); in BFCAllocator()121 bytes = RoundedBytes(bytes * kBackpedalFactor); in Extend()246 size_t BFCAllocator::RoundedBytes(size_t bytes) { in RoundedBytes() function in tensorflow::BFCAllocator265 size_t rounded_bytes = RoundedBytes(num_bytes); in AllocateRawInternal()
324 static size_t RoundedBytes(size_t bytes);
531 EXPECT_EQ(GPUBFCAllocator::RoundedBytes(size_t{1048576}), in TestForceAllowGrowth()542 EXPECT_EQ(GPUBFCAllocator::RoundedBytes(size_t{1048576}), in TestForceAllowGrowth()554 EXPECT_EQ(GPUBFCAllocator::RoundedBytes(size_t{1048576}), in TestForceAllowGrowth()566 EXPECT_EQ(GPUBFCAllocator::RoundedBytes(1LL << 31), in TestForceAllowGrowth()