Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/
Dbfc_allocator.cc43 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::BFCAllocator
265 size_t rounded_bytes = RoundedBytes(num_bytes); in AllocateRawInternal()
Dbfc_allocator.h324 static size_t RoundedBytes(size_t bytes);
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_bfc_allocator_test.cc531 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()