Searched refs:shape_layout (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | layout_assignment.h | 94 OperandLayoutConstraint(const ShapeLayout& shape_layout,
98 const ShapeLayout& shape_layout() const { return shape_layout_; } in shape_layout() function
116 explicit ResultLayoutConstraint(const ShapeLayout& shape_layout,
119 shape_layout_(shape_layout) {} in LayoutConstraint()
121 const ShapeLayout& shape_layout() const { return shape_layout_; } in shape_layout() function
354 Status PropagateUseConstraintToDefs(const ShapeLayout& shape_layout,
|
| D | computation_layout.h | 46 void add_parameter_layout(ShapeLayout shape_layout) { in add_parameter_layout() argument
47 parameter_layouts_.push_back(std::move(shape_layout)); in add_parameter_layout()
|
| D | layout_assignment.cc | 78 const ShapeLayout& shape_layout, const HloInstruction* instruction, in OperandLayoutConstraint() argument
81 shape_layout_(shape_layout), in OperandLayoutConstraint()
85 CHECK(ShapeUtil::Compatible(shape_layout.shape(), in OperandLayoutConstraint()
87 << shape_layout.shape() << " is not compatible with " in OperandLayoutConstraint()
209 if (curr_shape_layout->shape_layout().MatchesLayoutInShape( in SetOperandLayout()
219 curr_shape_layout->shape_layout().ToString(), in SetOperandLayout()
339 return &constraint->shape_layout(); in OperandLayout()
351 return result_constraint_ ? &result_constraint_->shape_layout() : nullptr; in ResultLayout()
1195 const ShapeLayout& shape_layout, const HloInstruction* instruction, in PropagateUseConstraintToDefs() argument
1202 [&shape_layout, constraints]( in PropagateUseConstraintToDefs()
[all …]
|
| D | layout_assignment_test.cc | 109 const ShapeLayout shape_layout(shape); in TEST_F() local
112 *computation_layout.mutable_parameter_layout(0) = shape_layout; in TEST_F()
113 *computation_layout.mutable_parameter_layout(1) = shape_layout; in TEST_F()
114 *computation_layout.mutable_result_layout() = shape_layout; in TEST_F()
191 const ShapeLayout shape_layout(shape); in TEST_F() local
194 *computation_layout.mutable_result_layout() = shape_layout; in TEST_F()
|
| D | BUILD | 742 "//tensorflow/compiler/xla:shape_layout",
778 "//tensorflow/compiler/xla:shape_layout",
2609 "//tensorflow/compiler/xla:shape_layout",
2847 "//tensorflow/compiler/xla:shape_layout",
3231 "//tensorflow/compiler/xla:shape_layout",
3246 "//tensorflow/compiler/xla:shape_layout",
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_layout_assignment.cc | 286 layout_constraint.shape_layout().shape(), instruction)); in PropagateOperandConstraint()
301 layout_constraint.shape_layout().layout(), *out_buf)); in PropagateOperandConstraint()
316 layout_constraint.shape_layout().layout(), *out_buf)); in PropagateOperandConstraint()
320 layout_constraint.shape_layout().shape(), instruction, operand_to_set)); in PropagateOperandConstraint()
|
| D | BUILD | 677 "//tensorflow/compiler/xla:shape_layout",
963 "//tensorflow/compiler/xla:shape_layout",
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | BUILD | 714 name = "shape_layout",
715 srcs = ["shape_layout.cc"],
716 hdrs = ["shape_layout.h"],
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | BUILD | 812 "//tensorflow/compiler/xla:shape_layout",
907 "//tensorflow/compiler/xla:shape_layout",
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | BUILD | 119 "//tensorflow/compiler/xla:shape_layout",
|