Searched refs:memory_limit_ (Results 1 – 8 of 8) 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::__anon298b6a7d0111::Buffer
|
| D | conv_ops_gpu.h | 55 : memory_limit_(memory_limit), total_byte_size_(0), context_(context) {} in DnnScratchAllocator()
56 int64 GetMemoryLimitInBytes() override { return memory_limit_; } in GetMemoryLimitInBytes()
64 if (byte_size > memory_limit_) { in AllocateBytes()
68 memory_limit_, ").")}; in AllocateBytes()
92 int64 memory_limit_;
|
| D | map_stage_op.cc | 97 std::size_t memory_limit_ TF_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()
240 if (has_memory_limit() && bytes > memory_limit_) { in check_memory_limit()
243 "' bytes into Staging Area with a memory limit of '", memory_limit_, in check_memory_limit()
344 memory_limit_(memory_limit), in StagingMap()
|
| D | fft_ops.cc | 366 : memory_limit_(memory_limit), total_byte_size_(0), context_(context) {} in CufftScratchAllocator()
367 int64 GetMemoryLimitInBytes() override { return memory_limit_; } in GetMemoryLimitInBytes()
371 if (byte_size > memory_limit_) { in AllocateBytes()
393 int64 memory_limit_; member in tensorflow::__anon049b3f3c0111::CufftScratchAllocator
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | redzone_allocator.h | 53 int64 GetMemoryLimitInBytes() override { return memory_limit_; } in GetMemoryLimitInBytes()
105 const int64 memory_limit_; variable
|
| D | redzone_allocator.cc | 55 memory_limit_(memory_limit), in RedzoneAllocator()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | bfc_allocator.cc | 59 memory_limit_ = total_memory; in BFCAllocator()
107 size_t available_bytes = memory_limit_ - total_region_allocated_bytes_; in Extend()
332 memory_limit_ - total_region_allocated_bytes_ + total_free_bytes; in DeallocateFreeRegions()
497 memory_limit_ - stats_.bytes_reserved - stats_.bytes_in_use; in AddTraceMe()
1059 << " memory_limit_: " << memory_limit_ << " available bytes: " in DumpMemoryLog()
1060 << (memory_limit_ - total_region_allocated_bytes_) in DumpMemoryLog()
|
| D | bfc_allocator.h | 523 size_t memory_limit_ = 0; variable
|