Searched refs:HloExecutionProfiler (Results 1 – 6 of 6) sorted by relevance
53 HloExecutionProfiler::HloExecutionProfiler( in HloExecutionProfiler() function in xla::gpu::HloExecutionProfiler67 void HloExecutionProfiler::FinishExecution() { in FinishExecution()77 void HloExecutionProfiler::StartHloComputation() { in StartHloComputation()83 void HloExecutionProfiler::FinishHloComputation( in FinishHloComputation()92 void HloExecutionProfiler::FinishHloComputation( in FinishHloComputation()101 void HloExecutionProfiler::StartHloInstruction() { in StartHloInstruction()107 void HloExecutionProfiler::FinishHloInstruction(size_t index) { in FinishHloInstruction()116 HloExecutionProfiler::MakeScopedInstructionProfiler( in MakeScopedInstructionProfiler()
37 class HloExecutionProfiler {40 explicit HloExecutionProfiler(bool do_profile, HloExecutionProfile* profile,96 ScopedInstructionProfiler(HloExecutionProfiler* profiler, in ScopedInstructionProfiler()110 HloExecutionProfiler* profiler_;
84 HloExecutionProfiler* profiler = nullptr,
109 HloExecutionProfiler* profiler; // never null
176 HloExecutionProfiler *profiler, in RunGemm()
158 HloExecutionProfiler profiler(do_profile, hlo_execution_profile, main_stream, in ExecuteThunks()