Home
last modified time | relevance | path

Searched refs:timing_counter_ (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device_test.cc283 timing_counter_.reset(new SharedCounter); in SetUp()
285 new GPUKernelTracker(Env::Default(), timing_counter_.get())); in SetUp()
289 std::unique_ptr<SharedCounter> timing_counter_; member in tensorflow::GPUKernelTrackerTest
312 EXPECT_EQ(timing_counter_->get(), kernel_tracker_->LastTerminatedCount()); in TEST_F()
317 int64 lower_bound = timing_counter_->get(); in TEST_F()
320 int64 upper_bound = timing_counter_->get(); in TEST_F()
341 EXPECT_EQ(timing_counter_->get(), kernel_tracker_->LastTerminatedCount()); in TEST_F()
Dgpu_device.h191 : env_(env), timing_counter_(timing_counter), pending_kernels_(64) { in GPUKernelTracker()
192 if (!timing_counter_) { in GPUKernelTracker()
196 timing_counter_ = owned_counter_.get(); in GPUKernelTracker()
232 SharedCounter* timing_counter_; variable
Dgpu_device.cc1698 uint64 queued_count = timing_counter_->next(); in RecordQueued()
/external/tensorflow/tensorflow/core/common_runtime/
Dbfc_allocator.h76 void SetTimingCounter(SharedCounter* sc) { timing_counter_ = sc; } in SetTimingCounter()
440 SharedCounter* timing_counter_ = nullptr; variable
Dbfc_allocator.cc495 if (timing_counter_) { in FreeAndMaybeCoalesce()
496 c->freed_count = timing_counter_->next(); in FreeAndMaybeCoalesce()
670 << (timing_counter_ in DumpMemoryLog()