Searched refs:cupti_hook_ (Results 1 – 1 of 1) sorted by relevance
405 std::unique_ptr<CuptiCallbackHook> cupti_hook_; member in tensorflow::DeviceTracerImpl430 cupti_hook_.reset(new CuptiCallbackHook()); in Start()431 TF_RETURN_IF_ERROR(cupti_hook_->Enable(recorder_.get())); in Start()447 cupti_hook_.reset(); in Stop()