Searched refs:ResultLayout (Results 1 – 2 of 2) sorted by relevance
267 const ShapeLayout* curr_shape_layout = ResultLayout(); in SetResultLayout()350 const ShapeLayout* LayoutConstraints::ResultLayout() const { in ResultLayout() function in xla::LayoutConstraints375 if (ResultLayout() != nullptr) { in ToString()376 absl::StrAppend(&output, " => ", ResultLayout()->ToString(), "\n"); in ToString()1796 if (constraints.ResultLayout() != nullptr && in RunOnComputation()1797 !constraints.ResultLayout()->MatchesLayoutInShape( in RunOnComputation()1801 CreateCopyWithNewLayout(constraints.ResultLayout()->shape(), in RunOnComputation()
163 const ShapeLayout* ResultLayout() const;