Home
last modified time | relevance | path

Searched refs:kernel_tracker_ (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device_test.cc284 kernel_tracker_.reset( in SetUp()
288 std::unique_ptr<GPUKernelTracker> kernel_tracker_; member in tensorflow::GPUKernelTrackerTest
293 EXPECT_EQ(0, kernel_tracker_->NumPending()); in TEST_F()
295 EXPECT_EQ(1, kernel_tracker_->LastTerminatedCount()); in TEST_F()
299 queued_counts.push_back(kernel_tracker_->RecordQueued()); in TEST_F()
301 EXPECT_EQ(32, kernel_tracker_->NumPending()); in TEST_F()
302 EXPECT_EQ(1, kernel_tracker_->LastTerminatedCount()); in TEST_F()
308 kernel_tracker_->RecordTerminated(x); in TEST_F()
309 EXPECT_EQ(queued_counts.size(), kernel_tracker_->NumPending()); in TEST_F()
310 EXPECT_EQ(x, kernel_tracker_->LastTerminatedCount()); in TEST_F()
[all …]
Dgpu_device.cc330 kernel_tracker_.reset(new GPUKernelTracker(Env::Default(), timing_counter)); in BaseGPUDevice()
565 DCHECK(kernel_tracker_); in ComputeHelper()
566 kernel_tracker_->PauseWhilePendingExceeds(pending_cap_); in ComputeHelper()
585 if (kernel_tracker_) { in ComputeHelper()
586 GPUKernelTracker* tracker = kernel_tracker_.get(); in ComputeHelper()
1683 return kernel_tracker_->LastTerminatedCount(); in SafeAllocFrontier()
1690 if (kernel_tracker_) { in PendingKernels()
1691 return kernel_tracker_->NumPending(); in PendingKernels()
Dgpu_device.h157 std::unique_ptr<GPUKernelTracker> kernel_tracker_; variable