Searched refs:FinishHloComputation (Results 1 – 5 of 5) sorted by relevance
54 void FinishHloComputation(const HloComputation* computation);59 void FinishHloComputation(absl::optional<size_t> profile_index);
68 profiler.FinishHloComputation(condition_profile_index_); in ExecuteOnStream()92 profiler.FinishHloComputation(body_profile_index_); in ExecuteOnStream()
83 void HloExecutionProfiler::FinishHloComputation( in FinishHloComputation() function in xla::gpu::HloExecutionProfiler92 void HloExecutionProfiler::FinishHloComputation( in FinishHloComputation() function in xla::gpu::HloExecutionProfiler
52 params.profiler->FinishHloComputation(body_profile_index_); in ExecuteOnStream()
105 profiler.FinishHloComputation(config_.branch_profile_indices[branch_index]); in ExecuteOnStream()