Searched refs:branch_computation_layouts (Results 1 – 1 of 1) sorted by relevance
885 absl::Span<const ComputationLayout> branch_computation_layouts) { in CheckConditionalLayout() argument889 branch_computation_layouts[0].result_layout().MatchesLayoutInShape( in CheckConditionalLayout()890 branch_computation_layouts[j].result_layout().shape(), in CheckConditionalLayout()893 branch_computation_layouts[j].result_layout().MatchesLayoutInShape( in CheckConditionalLayout()896 branch_computation_layouts[j].result_layout().MatchesLayoutInShape( in CheckConditionalLayout()902 branch_computation_layouts[j].parameter_layout(0).MatchesLayoutInShape( in CheckConditionalLayout()1209 std::vector<ComputationLayout> branch_computation_layouts; in CheckLayouts() local1211 branch_computation_layouts.reserve(branch_computations.size()); in CheckLayouts()1213 branch_computation_layouts.emplace_back( in CheckLayouts()1218 instruction, absl::MakeSpan(branch_computation_layouts))); in CheckLayouts()