Searched refs:body_layout (Results 1 – 1 of 1) sorted by relevance
572 ComputationLayout& body_layout = FindOrDie(computation_layouts_, body); in AddMandatoryConstraints() local580 DCHECK(ShapeUtil::Compatible(body_layout.result_shape(), in AddMandatoryConstraints()581 body_layout.parameter_shape(0))); in AddMandatoryConstraints()582 DCHECK(ShapeUtil::Compatible(body_layout.result_shape(), in AddMandatoryConstraints()584 DCHECK(ShapeUtil::Compatible(body_layout.result_shape(), init->shape())); in AddMandatoryConstraints()586 if (body_layout.result_layout() != body_layout.parameter_layout(0)) { in AddMandatoryConstraints()589 << " shape=" << body_layout.result_layout().ToString(); in AddMandatoryConstraints()590 *body_layout.mutable_parameter_layout(0) = body_layout.result_layout(); in AddMandatoryConstraints()593 body_layout.parameter_layout(0)) { in AddMandatoryConstraints()596 << " shape=" << body_layout.parameter_layout(0).ToString(); in AddMandatoryConstraints()[all …]