Home
last modified time | relevance | path

Searched refs:ShapeLayout (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dshape_layout.cc25 Status ShapeLayout::CopyLayoutFromShape(const Shape& other_shape) { in CopyLayoutFromShape()
35 Status ShapeLayout::AssignLayoutToShape(Shape* to_shape) const { in AssignLayoutToShape()
45 void ShapeLayout::SetToDefaultLayout() { in SetToDefaultLayout()
49 bool ShapeLayout::MatchesLayoutInShape(const Shape& shape) const { in MatchesLayoutInShape()
53 const Layout& ShapeLayout::layout() const { in layout()
59 void ShapeLayout::Clear() { LayoutUtil::ClearLayout(&shape_); } in Clear()
61 bool ShapeLayout::LayoutIsSet() const { return LayoutUtil::HasLayout(shape_); } in LayoutIsSet()
63 void ShapeLayout::ResetLayout(const Layout& layout) { in ResetLayout()
70 void ShapeLayout::ResetLayout(const Layout& layout, in ResetLayout()
78 bool ShapeLayout::operator==(const ShapeLayout& other) const { in operator ==()
[all …]
Dshape_layout.h34 class ShapeLayout {
38 explicit ShapeLayout(const Shape& shape) : shape_(shape) {} in ShapeLayout() function
83 bool operator==(const ShapeLayout& other) const;
84 bool operator!=(const ShapeLayout& other) const;
/external/tensorflow/tensorflow/compiler/xla/service/
Dcomputation_layout.h36 explicit ComputationLayout(ShapeLayout result_layout) in ComputationLayout()
46 void add_parameter_layout(ShapeLayout shape_layout) { in add_parameter_layout()
51 const ShapeLayout& parameter_layout(int64 param_no) const { in parameter_layout()
54 ShapeLayout* mutable_parameter_layout(int64 param_no) { in mutable_parameter_layout()
62 const std::vector<ShapeLayout>& parameter_layouts() const { in parameter_layouts()
67 const ShapeLayout& result_layout() const { return result_layout_; } in result_layout()
68 ShapeLayout* mutable_result_layout() { return &result_layout_; } in mutable_result_layout()
91 std::vector<ShapeLayout> parameter_layouts_;
92 ShapeLayout result_layout_;
Dlayout_assignment.h94 OperandLayoutConstraint(const ShapeLayout& shape_layout,
98 const ShapeLayout& shape_layout() const { return shape_layout_; } in shape_layout()
108 ShapeLayout shape_layout_;
116 explicit ResultLayoutConstraint(const ShapeLayout& shape_layout,
121 const ShapeLayout& shape_layout() const { return shape_layout_; } in shape_layout()
125 const ShapeLayout shape_layout_;
159 const ShapeLayout* OperandLayout(const HloInstruction* instruction,
163 const ShapeLayout* ResultLayout() const;
354 Status PropagateUseConstraintToDefs(const ShapeLayout& shape_layout,
469 Status CopyOperandIfLayoutsDiffer(const ShapeLayout& operand_layout,
Dlayout_assignment_test.cc109 const ShapeLayout shape_layout(shape); in TEST_F()
139 const ShapeLayout col_major(col_major_shape); in TEST_F()
144 const ShapeLayout row_major(row_major_shape); in TEST_F()
191 const ShapeLayout shape_layout(shape); in TEST_F()
373 ShapeLayout(ashape_with_layout); in TEST_F()
374 *computation_layout.mutable_result_layout() = ShapeLayout(bshape_with_layout); in TEST_F()
414 ShapeLayout(ashape_with_layout); in TEST_F()
415 *computation_layout.mutable_result_layout() = ShapeLayout(bshape_with_layout); in TEST_F()
452 ShapeLayout(input_shape_with_layout); in TEST_F()
454 ShapeLayout(output_shape_with_layout); in TEST_F()
[all …]
Dlayout_assignment.cc78 const ShapeLayout& shape_layout, const HloInstruction* instruction, in OperandLayoutConstraint()
238 key, OperandLayoutConstraint(ShapeLayout(shape_with_layout), in SetOperandLayout()
243 OperandLayoutConstraint(ShapeLayout(shape_with_layout), instruction, in SetOperandLayout()
267 const ShapeLayout* curr_shape_layout = ResultLayout(); in SetResultLayout()
281 new ResultLayoutConstraint(ShapeLayout(shape_with_layout), dfs)); in SetResultLayout()
335 const ShapeLayout* LayoutConstraints::OperandLayout( in OperandLayout()
350 const ShapeLayout* LayoutConstraints::ResultLayout() const { in ResultLayout()
457 const ShapeLayout& parameter_layout = in AddMandatoryConstraints()
644 const ShapeLayout& result_layout = computation_layout->result_layout(); in AddMandatoryConstraints()
746 const ShapeLayout& parameter_layout = in CheckParameterLayout()
[all …]
Dhlo_module_config.cc45 for (const ShapeLayout& param_layout : in compilation_cache_key()
Dcomputation_layout.cc46 [](const ShapeLayout& s) { return s.LayoutIsSet(); }) && in LayoutIsSet()
Dhlo_computation.cc135 ShapeLayout(instruction->shape())); in AddEntryComputationParameter()
Dhlo_verifier.cc945 const ShapeLayout& result_layout = layout.result_layout(); in VerifyEntryComputationLayout()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_layout_assignment_test.cc81 ShapeLayout(LayoutUtil::GetWithDefaultLayout(lhs_shape)); in TEST_F()
83 ShapeLayout(LayoutUtil::GetWithDefaultLayout(result_shape)); in TEST_F()
122 ShapeLayout(LayoutUtil::GetWithDefaultLayout(lhs_shape)); in TEST_F()
124 ShapeLayout(LayoutUtil::GetWithDefaultLayout(result_shape)); in TEST_F()
166 ShapeLayout(LayoutUtil::GetWithDefaultLayout(lhs_a_shape)); in TEST_F()
168 ShapeLayout(LayoutUtil::GetWithDefaultLayout(lhs_b_shape)); in TEST_F()
170 ShapeLayout(LayoutUtil::GetWithDefaultLayout(tuple_result->shape())); in TEST_F()
200 ShapeLayout(LayoutUtil::GetWithDefaultLayout(rhs_shape)); in TEST_F()
202 ShapeLayout(LayoutUtil::GetWithDefaultLayout(result_shape)); in TEST_F()
240 ShapeLayout(LayoutUtil::GetWithDefaultLayout(lhs_shape)); in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment_test.cc71 ShapeLayout(lhs_shape_with_layout); in TEST_F()
73 ShapeLayout(rhs_shape_with_layout); in TEST_F()
75 ShapeLayout(result_shape_with_layout); in TEST_F()
159 ShapeLayout(input_shape); in TEST_F()
163 *computation_layout.mutable_result_layout() = ShapeLayout(result_shape); in TEST_F()
228 ShapeLayout(input_shape); in TEST_F()
233 ShapeLayout(ShapeUtil::MakeTupleShape( in TEST_F()
310 ShapeLayout(input_shape); in TEST_F()
315 ShapeLayout(ShapeUtil::MakeTupleShape( in TEST_F()