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.h51 class LayoutConstraint {
53 LayoutConstraint(bool mandatory, bool dfs) in LayoutConstraint() function
55 virtual ~LayoutConstraint() = default;
70 std::ostream& operator<<(std::ostream& out, const LayoutConstraint& constraint);
74 class BufferLayoutConstraint : public LayoutConstraint {
94 class OperandLayoutConstraint : public LayoutConstraint {
116 class ResultLayoutConstraint : public LayoutConstraint {
120 : LayoutConstraint(/*mandatory=*/true, dfs), in LayoutConstraint() function
148 std::vector<const LayoutConstraint*> ConsumeAddedConstraints() { in ConsumeAddedConstraints()
149 std::vector<const LayoutConstraint*> ret_vec(std::move(added_constraints_)); in ConsumeAddedConstraints()
[all …]
Dlayout_assignment.cc65 const LayoutConstraint& constraint) { in operator <<()
73 : LayoutConstraint(mandatory, dfs), layout_(layout), buffer_(&buffer) { in BufferLayoutConstraint()
85 : LayoutConstraint(mandatory, dfs), in OperandLayoutConstraint()
1179 std::deque<const LayoutConstraint*> worklist; in PropagateConstraints()
1195 const LayoutConstraint* layout_constraint = worklist.front(); in PropagateConstraints()