Searched refs:GPUKernelTracker (Results 1 – 4 of 4) sorted by relevance
50 class GPUKernelTracker; variable153 std::unique_ptr<GPUKernelTracker> kernel_tracker_;179 class GPUKernelTracker {201 explicit GPUKernelTracker(const Params& params, Env* env, in GPUKernelTracker() function
465 void Init(const GPUKernelTracker::Params& params) { in Init()467 kernel_tracker_.reset(new GPUKernelTracker(params, Env::Default(), nullptr, in Init()477 std::unique_ptr<GPUKernelTracker> kernel_tracker_;
447 GPUKernelTracker::Params tracker_params( in Init()467 kernel_tracker_.reset(new GPUKernelTracker( in Init()586 GPUKernelTracker* tracker = kernel_tracker_.get(); in Compute()1933 uint64 GPUKernelTracker::MaybeQueue(OpKernelContext* ctx) { in MaybeQueue()1959 void GPUKernelTracker::RecordQueued(uint64 queued_count, int weight) { in RecordQueued()2006 void GPUKernelTracker::MaybeQueueProgressEvent() { in MaybeQueueProgressEvent()2016 void GPUKernelTracker::RecordTerminated(uint64 queued_count) { in RecordTerminated()
182 // Parameters for GPUKernelTracker. By default no kernel tracking is done.