Searched refs:GPUKernelTracker (Results 1 – 4 of 4) sorted by relevance
55 class GPUKernelTracker; variable182 std::unique_ptr<GPUKernelTracker> kernel_tracker_;213 class GPUKernelTracker {235 explicit GPUKernelTracker(const Params& params, Env* env, in GPUKernelTracker() function
466 GPUKernelTracker::Params tracker_params( in Init()486 kernel_tracker_.reset(new GPUKernelTracker( in Init()705 GPUKernelTracker* tracker = kernel_tracker_.get(); in Compute()2066 uint64 GPUKernelTracker::MaybeQueue(OpKernelContext* ctx) { in MaybeQueue()2092 void GPUKernelTracker::RecordQueued(uint64 queued_count, int weight) { in RecordQueued()2139 void GPUKernelTracker::MaybeQueueProgressEvent() { in MaybeQueueProgressEvent()2149 void GPUKernelTracker::RecordTerminated(uint64 queued_count) { in RecordTerminated()
577 void Init(const GPUKernelTracker::Params& params) { in Init()579 kernel_tracker_.reset(new GPUKernelTracker(params, Env::Default(), nullptr, in Init()589 std::unique_ptr<GPUKernelTracker> kernel_tracker_;
204 // Parameters for GPUKernelTracker. By default no kernel tracking is done.