HomeSort by relevance Sort by last modified time
    Searched defs:memory_limit_ (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
conv_ops_gpu.h 43 : memory_limit_(memory_limit), total_byte_size_(0), context_(context) {}
45 return memory_limit_;
55 if (byte_size > memory_limit_) {
80 int64 memory_limit_; member in class:tensorflow::CudnnScratchAllocator
stage_op.cc 41 std::size_t memory_limit_; member in class:tensorflow::__anon39937::Buffer
65 bool IsBounded() const { return capacity_ > 0 || memory_limit_ > 0; }
70 return bytes + current_bytes_ > memory_limit_;
83 : capacity_(capacity), memory_limit_(memory_limit), current_bytes_(0) {}
92 if (memory_limit_ > 0 && tuple_bytes > memory_limit_) {
99 memory_limit_, "'."));
107 memory_limit_ > 0 ? !WouldExceedMemoryLimit(tuple_bytes) : true;
fft_ops.cc 306 : memory_limit_(memory_limit), total_byte_size_(0), context_(context) {}
308 return memory_limit_;
313 if (byte_size > memory_limit_) {
335 int64 memory_limit_; member in class:tensorflow::__anon39800::CufftScratchAllocator
  /external/tensorflow/tensorflow/core/common_runtime/
bfc_allocator.h 366 size_t memory_limit_ = 0; member in class:tensorflow::BFCAllocator

Completed in 115 milliseconds