Searched refs:result_layout_ (Results 1 – 4 of 4) sorted by relevance
37 : result_layout_(std::move(result_layout)) {} in ComputationLayout()67 const ShapeLayout& result_layout() const { return result_layout_; } in result_layout()68 ShapeLayout* mutable_result_layout() { return &result_layout_; } in mutable_result_layout()75 const Shape& result_shape() const { return result_layout_.shape(); } in result_shape()92 ShapeLayout result_layout_; variable
28 : result_layout_(program_shape.result()) { in ComputationLayout()41 result_layout_.SetToDefaultLayout(); in SetToDefaultLayout()47 result_layout_.LayoutIsSet(); in LayoutIsSet()56 result_layout_.ToString()); in ToString()65 *program_shape.mutable_result() = result_layout_.shape(); in ComputeProgramShape()
53 result_layout_ = shape_with_layout; in set_result_layout()58 return result_layout_set_ ? &result_layout_ : nullptr; in result_layout()70 result_layout = ShapeUtil::HumanStringWithLayout(result_layout_); in ToString()
77 Shape result_layout_; variable