Searched refs:kernel_tracker_ (Results 1 – 3 of 3) sorted by relevance
467 kernel_tracker_.reset(new GPUKernelTracker(params, Env::Default(), nullptr, in Init()473 mutex_lock l(kernel_tracker_->mu_); in RecordQueued()474 kernel_tracker_->RecordQueued(v, 1); in RecordQueued()477 std::unique_ptr<GPUKernelTracker> kernel_tracker_; member in tensorflow::GPUKernelTrackerTest483 EXPECT_EQ(0, kernel_tracker_->NumPending()); in TEST_F()485 EXPECT_EQ(1, kernel_tracker_->LastTerminatedCount(0)); in TEST_F()493 EXPECT_EQ(32, kernel_tracker_->NumPending()); in TEST_F()494 EXPECT_EQ(1, kernel_tracker_->LastTerminatedCount(0)); in TEST_F()500 kernel_tracker_->RecordTerminated(x); in TEST_F()501 EXPECT_EQ(queued_counts.size(), kernel_tracker_->NumPending()); in TEST_F()[all …]
467 kernel_tracker_.reset(new GPUKernelTracker( in Init()560 if (kernel_tracker_.get()) { in Compute()563 kernel_tracker_->PauseWhilePendingExceeds(pending_cap_); in Compute()585 if (kernel_tracker_) { in Compute()586 GPUKernelTracker* tracker = kernel_tracker_.get(); in Compute()1916 return kernel_tracker_->LastTerminatedCount(old_value); in SafeAllocFrontier()1923 if (kernel_tracker_) { in PendingKernels()1924 return kernel_tracker_->NumPending(); in PendingKernels()
153 std::unique_ptr<GPUKernelTracker> kernel_tracker_; variable