Searched refs:ResultLayout (Results 1 – 2 of 2) sorted by relevance
277 const ShapeLayout* curr_shape_layout = ResultLayout(); in SetResultLayout()360 const ShapeLayout* LayoutConstraints::ResultLayout() const { in ResultLayout() function in xla::LayoutConstraints385 if (ResultLayout() != nullptr) { in ToString()386 absl::StrAppend(&output, " => ", ResultLayout()->ToString(), "\n"); in ToString()1936 if (constraints.ResultLayout() != nullptr) { in RunOnComputation()1939 if (!constraints.ResultLayout()->MatchesLayoutInShape( in RunOnComputation()1948 CreateCopyWithNewLayout(constraints.ResultLayout()->shape(), in RunOnComputation()1957 constraints.ResultLayout()->shape(), index); in RunOnComputation()
165 const ShapeLayout* ResultLayout() const;