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.cc52 curr_region_allocation_bytes_ = in BFCAllocator()
55 curr_region_allocation_bytes_ = RoundedBytes(total_memory); in BFCAllocator()
121 while (rounded_bytes > curr_region_allocation_bytes_) { in Extend()
122 curr_region_allocation_bytes_ *= 2; in Extend()
127 size_t bytes = std::min(curr_region_allocation_bytes_, available_bytes); in Extend()
150 curr_region_allocation_bytes_ *= 2; in Extend()
1062 << curr_region_allocation_bytes_; in DumpMemoryLog()
Dbfc_allocator.h564 size_t curr_region_allocation_bytes_; variable
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_bfc_allocator_test.cc555 unset_flag_allocator.curr_region_allocation_bytes_); in TestForceAllowGrowth()
563 unparsable_flag_allocator.curr_region_allocation_bytes_); in TestForceAllowGrowth()
572 force_allow_growth_allocator.curr_region_allocation_bytes_); in TestForceAllowGrowth()
581 force_no_allow_growth_allocator.curr_region_allocation_bytes_); in TestForceAllowGrowth()