Searched refs:FinishHloComputation (Results 1 – 5 of 5) sorted by relevance
63 profiler->FinishHloComputation(hlo_instruction()->while_condition()); in ExecuteOnStream()88 profiler->FinishHloComputation(hlo_instruction()->while_body()); in ExecuteOnStream()
54 profiler->FinishHloComputation(hlo_instruction()->while_body()); in ExecuteOnStream()
55 void FinishHloComputation(const HloComputation* computation);
93 profiler->FinishHloComputation( in ExecuteOnStream()
85 void HloExecutionProfiler::FinishHloComputation( in FinishHloComputation() function in xla::gpu::HloExecutionProfiler