Searched refs:cupti_wrapper_ (Results 1 – 1 of 1) sorted by relevance
168 : cupti_wrapper_(new perftools::gputools::profiler::CuptiWrapper()), in CuptiCallbackHook()173 cupti_wrapper_->Subscribe(&subscriber_, &CuptiCallback, recorder))); in Enable()183 TF_RETURN_IF_ERROR(ToStatus(cupti_wrapper_->EnableCallback( in Enable()190 LogIfError(ToStatus(cupti_wrapper_->Unsubscribe(subscriber_))); in ~CuptiCallbackHook()315 std::unique_ptr<perftools::gputools::profiler::CuptiWrapper> cupti_wrapper_; member in tensorflow::__anonee34c7620111::CuptiCallbackHook