Searched refs:memory_limit_ (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | stage_op.cc | 38 : capacity_(capacity), memory_limit_(memory_limit), current_bytes_(0) {} in Buffer()
47 if (memory_limit_ > 0 && tuple_bytes > memory_limit_) { in Put()
54 memory_limit_, "'.")); in Put()
62 memory_limit_ > 0 ? !WouldExceedMemoryLimit(tuple_bytes) : true; in Put()
155 bool IsBounded() const { return capacity_ > 0 || memory_limit_ > 0; } in IsBounded()
160 return bytes + current_bytes_ > memory_limit_; in WouldExceedMemoryLimit()
171 std::size_t memory_limit_; member in tensorflow::__anon4b7f969f0111::Buffer
|
| D | conv_ops_gpu.h | 43 : memory_limit_(memory_limit), total_byte_size_(0), context_(context) {} in DnnScratchAllocator()
45 return memory_limit_; in GetMemoryLimitInBytes()
54 if (byte_size > memory_limit_) { in AllocateBytes()
76 int64 memory_limit_;
|
| D | map_stage_op.cc | 97 std::size_t memory_limit_ GUARDED_BY(mu_);
133 return memory_limit_ > 0; in has_memory_limit()
138 return has_memory_limit() && bytes + current_bytes_ > memory_limit_; in would_exceed_memory_limit()
238 if (has_memory_limit() && bytes > memory_limit_) { in check_memory_limit()
241 "' bytes into Staging Area with a memory limit of '", memory_limit_, in check_memory_limit()
342 memory_limit_(memory_limit), in StagingMap()
|
| D | fft_ops.cc | 315 : memory_limit_(memory_limit), total_byte_size_(0), context_(context) {} in CufftScratchAllocator()
317 return memory_limit_; in GetMemoryLimitInBytes()
322 if (byte_size > memory_limit_) { in AllocateBytes()
344 int64 memory_limit_; member in tensorflow::__anona0c96f1e0111::CufftScratchAllocator
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | bfc_allocator.cc | 49 memory_limit_ = total_memory; in BFCAllocator()
91 size_t available_bytes = memory_limit_ - total_region_allocated_bytes_; in Extend()
|
| D | bfc_allocator.h | 388 size_t memory_limit_ = 0; variable
|