Searched refs:branch_profile_indices (Results 1 – 3 of 3) sorted by relevance
30 std::vector<absl::optional<size_t>> branch_profile_indices) { in GetConditionalThunkConfig() argument43 config.branch_profile_indices = std::move(branch_profile_indices); in GetConditionalThunkConfig()105 profiler.FinishHloComputation(config_.branch_profile_indices[branch_index]); in ExecuteOnStream()
37 std::vector<absl::optional<size_t>> branch_profile_indices; member43 std::vector<absl::optional<size_t>> branch_profile_indices);
3902 std::vector<absl::optional<size_t>> branch_profile_indices; in BuildConditionalThunk() local3906 branch_profile_indices.reserve(branch_count); in BuildConditionalThunk()3924 branch_profile_indices.push_back(profile_index); in BuildConditionalThunk()3928 hlo, std::move(branch_thunks), std::move(branch_profile_indices)); in BuildConditionalThunk()