Searched refs:CudaEventCollector (Results 1 – 1 of 1) sorted by relevance
456 class CudaEventCollector { class479 CudaEventCollector(CudaEventRecorder* recorder, StepStatsCollector* collector) in CudaEventCollector() function in tensorflow::__anon18c3f2790211::CudaEventCollector693 auto status = CudaEventCollector(recorder, collector).Collect(); in Collect()715 TF_RETURN_IF_ERROR(CudaEventCollector::Collect(recorder_.get(), collector)); in Collect()