Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.h50 class GPUKernelTracker; variable
157 std::unique_ptr<GPUKernelTracker> kernel_tracker_;
185 class GPUKernelTracker {
190 explicit GPUKernelTracker(Env* env, SharedCounter* timing_counter) in GPUKernelTracker() function
Dgpu_device_test.cc285 new GPUKernelTracker(Env::Default(), timing_counter_.get())); in SetUp()
288 std::unique_ptr<GPUKernelTracker> kernel_tracker_;
Dgpu_device.cc330 kernel_tracker_.reset(new GPUKernelTracker(Env::Default(), timing_counter)); in BaseGPUDevice()
586 GPUKernelTracker* tracker = kernel_tracker_.get(); in ComputeHelper()
1696 uint64 GPUKernelTracker::RecordQueued() { in RecordQueued()
1730 void GPUKernelTracker::RecordTerminated(uint64 queued_count) { in RecordTerminated()
1778 uint64 GPUKernelTracker::LastTerminatedCount() { in LastTerminatedCount()