Searched refs:kernel_tracker_ (Results 1 – 3 of 3) sorted by relevance
284 kernel_tracker_.reset( in SetUp()288 std::unique_ptr<GPUKernelTracker> kernel_tracker_; member in tensorflow::GPUKernelTrackerTest293 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 …]
330 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()
157 std::unique_ptr<GPUKernelTracker> kernel_tracker_; variable