Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/platform/default/
Ddevice_tracer.cc404 std::unique_ptr<CudaEventRecorder> recorder_; member in tensorflow::DeviceTracerImpl
412 DeviceTracerImpl::DeviceTracerImpl() : recorder_(new CudaEventRecorder()) { in DeviceTracerImpl()
431 TF_RETURN_IF_ERROR(cupti_hook_->Enable(recorder_.get())); in Start()
480 : recorder_(recorder), collector_(collector) { in CudaEventCollector()
653 auto kernel_records = recorder_->ConsumeKernelRecords(); in Collect()
654 auto memcpy_records = recorder_->ConsumeMemcpyRecords(); in Collect()
699 CudaEventRecorder* recorder_; member in tensorflow::__anonee34c7620211::CudaEventCollector
715 TF_RETURN_IF_ERROR(CudaEventCollector::Collect(recorder_.get(), collector)); in Collect()