Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_process_state.cc118 SharedCounter* timing_counter = nullptr; in GetGPUAllocator() local
120 timing_counter = new SharedCounter; in GetGPUAllocator()
121 gpu_bfc_allocator->SetTimingCounter(timing_counter); in GetGPUAllocator()
148 std::unique_ptr<SharedCounter>(timing_counter), in GetGPUAllocator()
Dgpu_device.h190 explicit GPUKernelTracker(Env* env, SharedCounter* timing_counter) in GPUKernelTracker() argument
191 : env_(env), timing_counter_(timing_counter), pending_kernels_(64) { in GPUKernelTracker()
Dgpu_device.cc318 SharedCounter* timing_counter = nullptr; in BaseGPUDevice() local
324 timing_counter = in BaseGPUDevice()
326 DCHECK(timing_counter); in BaseGPUDevice()
330 kernel_tracker_.reset(new GPUKernelTracker(Env::Default(), timing_counter)); in BaseGPUDevice()