Home
last modified time | relevance | path

Searched refs:pending_cap_ (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.cc314 pending_cap_ = options.config.gpu_options().experimental().pending_cap(); in BaseGPUDevice()
317 if (timestamped_allocator_ || pending_cap_ > 0) { in BaseGPUDevice()
328 DCHECK_GT(pending_cap_, 0); in BaseGPUDevice()
564 if (pending_cap_ > 0) { in ComputeHelper()
566 kernel_tracker_->PauseWhilePendingExceeds(pending_cap_); in ComputeHelper()
Dgpu_device.h158 int pending_cap_ = 0; variable