Searched refs:SetResultLayout (Results 1 – 2 of 2) sorted by relevance
175 Status SetResultLayout(const Shape& shape_with_layout, bool dfs = true);
272 Status LayoutConstraints::SetResultLayout(const Shape& shape_with_layout, in SetResultLayout() function in xla::LayoutConstraints654 TF_RETURN_IF_ERROR(constraints->SetResultLayout( in AddMandatoryConstraints()659 TF_RETURN_IF_ERROR(constraints->SetResultLayout(result_layout.shape())); in AddMandatoryConstraints()