Searched refs:memory_limit_bytes_ (Results 1 – 6 of 6) sorted by relevance
116 memory_limit_bytes_ = bytes; in SetMemoryLimit()172 if (memory_limit_bytes_ == 0) in CheckMemory()178 if (memory_window_bytes_.Mean() > memory_limit_bytes_) { in CheckMemory()182 memory_window_bytes_.Mean(), memory_limit_bytes_); in CheckMemory()
63 memory_limit_bytes_(memory_limit_bytes), in HloRematerialization()106 const int64 memory_limit_bytes_; variable
1205 << HumanReadableNumBytes(memory_limit_bytes_); in Run()1230 memory_limit_bytes_ - module_output_size; in Run()1298 if (current_peak_memory > memory_limit_bytes_) { in Run()1302 HumanReadableNumBytes(memory_limit_bytes_), memory_limit_bytes_, in Run()
143 uint64_t memory_limit_bytes_ = 0; variable
191 memory_limit_bytes_(GetMemoryLimitBytes()) { in StreamExecutor()502 if (memory_limit_bytes_ > 0 && in Allocate()503 mem_alloc_bytes_ + size > memory_limit_bytes_) { in Allocate()507 << ", limit=" << memory_limit_bytes_ << "]"; in Allocate()
720 int64 memory_limit_bytes_; variable