Searched refs:body_layout (Results 1 – 1 of 1) sorted by relevance
554 ComputationLayout& body_layout = FindOrDie(computation_layouts_, body); in AddMandatoryConstraints() local562 DCHECK(ShapeUtil::Compatible(body_layout.result_shape(), in AddMandatoryConstraints()563 body_layout.parameter_shape(0))); in AddMandatoryConstraints()564 DCHECK(ShapeUtil::Compatible(body_layout.result_shape(), in AddMandatoryConstraints()566 DCHECK(ShapeUtil::Compatible(body_layout.result_shape(), init->shape())); in AddMandatoryConstraints()568 if (body_layout.result_layout() != body_layout.parameter_layout(0)) { in AddMandatoryConstraints()571 << " shape=" << body_layout.result_layout().ToString(); in AddMandatoryConstraints()572 *body_layout.mutable_parameter_layout(0) = body_layout.result_layout(); in AddMandatoryConstraints()575 body_layout.parameter_layout(0)) { in AddMandatoryConstraints()578 << " shape=" << body_layout.parameter_layout(0).ToString(); in AddMandatoryConstraints()[all …]