Searched refs:branch_computation_layout (Results 1 – 1 of 1) sorted by relevance
765 ComputationLayout branch_computation_layout = in AddMandatoryConstraints() local768 if (!branch_computation_layout.result_layout().MatchesLayoutInShape( in AddMandatoryConstraints()771 *branch_computation_layout.mutable_result_layout() = in AddMandatoryConstraints()775 branch_computation_layout); in AddMandatoryConstraints()778 branch_computation_layout.parameter_shape(0), instruction, k + 1, in AddMandatoryConstraints()