Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_bfc_allocator_test.cc554 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()
/external/tensorflow/tensorflow/core/common_runtime/
Dbfc_allocator.cc53 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::BFCAllocator
399 size_t rounded_bytes = RoundedBytes(num_bytes); in AllocateRawInternal()
Dbfc_allocator.h436 static size_t RoundedBytes(size_t bytes);