Home
last modified time | relevance | path

Searched refs:SharedCounter (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_process_state.h37 class SharedCounter; variable
113 SharedCounter* GPUAllocatorCounter(TfGpuId tf_gpu_id);
139 std::unique_ptr<SharedCounter> counter;
Dgpu_process_state.cc118 SharedCounter* timing_counter = nullptr; in GetGPUAllocator()
120 timing_counter = new SharedCounter; in GetGPUAllocator()
148 std::unique_ptr<SharedCounter>(timing_counter), in GetGPUAllocator()
164 SharedCounter* GPUProcessState::GPUAllocatorCounter(TfGpuId tf_gpu_id) { in GPUAllocatorCounter()
253 std::unique_ptr<SharedCounter>(nullptr), in GetGpuHostAllocator()
Dgpu_device.h190 explicit GPUKernelTracker(Env* env, SharedCounter* timing_counter) in GPUKernelTracker()
195 owned_counter_.reset(new SharedCounter); in GPUKernelTracker()
232 SharedCounter* timing_counter_;
233 std::unique_ptr<SharedCounter> owned_counter_;
Dgpu_device_test.cc283 timing_counter_.reset(new SharedCounter); in SetUp()
289 std::unique_ptr<SharedCounter> timing_counter_;
Dgpu_device.cc318 SharedCounter* timing_counter = nullptr; in BaseGPUDevice()
/external/tensorflow/tensorflow/core/common_runtime/
Dshared_counter.h21 class SharedCounter {
Dbfc_allocator.h76 void SetTimingCounter(SharedCounter* sc) { timing_counter_ = sc; } in SetTimingCounter()
440 SharedCounter* timing_counter_ = nullptr;