Home
last modified time | relevance | path

Searched refs:LayoutConstraint (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment.h49 class LayoutConstraint {
51 LayoutConstraint(bool mandatory, bool dfs) in LayoutConstraint() function
53 virtual ~LayoutConstraint() = default;
68 std::ostream& operator<<(std::ostream& out, const LayoutConstraint& constraint);
72 class BufferLayoutConstraint : public LayoutConstraint {
92 class OperandLayoutConstraint : public LayoutConstraint {
114 class ResultLayoutConstraint : public LayoutConstraint {
118 : LayoutConstraint(/*mandatory=*/true, dfs), in LayoutConstraint() function
146 std::vector<const LayoutConstraint*> ConsumeAddedConstraints() { in ConsumeAddedConstraints()
147 std::vector<const LayoutConstraint*> ret_vec(std::move(added_constraints_)); in ConsumeAddedConstraints()
[all …]
Dlayout_assignment.cc60 const LayoutConstraint& constraint) { in operator <<()
68 : LayoutConstraint(mandatory, dfs), layout_(layout), buffer_(&buffer) { in BufferLayoutConstraint()
80 : LayoutConstraint(mandatory, dfs), in OperandLayoutConstraint()
1121 std::deque<const LayoutConstraint*> worklist; in PropagateConstraints()
1137 const LayoutConstraint* layout_constraint = worklist.front(); in PropagateConstraints()