Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/platform/default/
Ddevice_tracer.cc107 class CudaEventRecorder { class
171 Status Enable(CudaEventRecorder* recorder) { in Enable()
198 auto recorder = static_cast<CudaEventRecorder*>(userdata); in CuptiCallback()
224 CudaEventRecorder* recorder) { in StartMemcpy()
232 CudaEventRecorder* recorder) { in StartMemcpyAsync()
240 CudaEventRecorder* recorder) { in DriverApiEnterCallback()
295 CudaEventRecorder* recorder) { in DriverApiExitCallback()
404 std::unique_ptr<CudaEventRecorder> recorder_;
412 DeviceTracerImpl::DeviceTracerImpl() : recorder_(new CudaEventRecorder()) { in DeviceTracerImpl()
479 CudaEventCollector(CudaEventRecorder* recorder, StepStatsCollector* collector) in CudaEventCollector()
[all …]