Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/
Dbfc_allocator.cc42 curr_region_allocation_bytes_ = in BFCAllocator()
45 curr_region_allocation_bytes_ = RoundedBytes(total_memory); in BFCAllocator()
105 while (rounded_bytes > curr_region_allocation_bytes_) { in Extend()
106 curr_region_allocation_bytes_ *= 2; in Extend()
111 size_t bytes = std::min(curr_region_allocation_bytes_, available_bytes); in Extend()
133 curr_region_allocation_bytes_ *= 2; in Extend()
Dbfc_allocator.h429 size_t curr_region_allocation_bytes_; variable
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_bfc_allocator_test.cc532 unset_flag_allocator.curr_region_allocation_bytes_); in TestForceAllowGrowth()
543 unparsable_flag_allocator.curr_region_allocation_bytes_); in TestForceAllowGrowth()
555 force_allow_growth_allocator.curr_region_allocation_bytes_); in TestForceAllowGrowth()
567 force_no_allow_growth_allocator.curr_region_allocation_bytes_); in TestForceAllowGrowth()