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