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,
56 virtual ~LayoutConstraint() = default;
79 std::ostream& operator<<(std::ostream& out, const LayoutConstraint& constraint);
83 class BufferLayoutConstraint : public LayoutConstraint {
87 int64_t priority = LayoutConstraint::kDefaultPriority);
104 class OperandLayoutConstraint : public LayoutConstraint {
109 int64_t priority = LayoutConstraint::kDefaultPriority);
127 class ResultLayoutConstraint : public LayoutConstraint {
131 int64_t priority = LayoutConstraint::kDefaultPriority)
[all …]
Dlayout_assignment.cc65 const LayoutConstraint& constraint) { in operator <<()
74 : LayoutConstraint(mandatory, dfs, priority), in BufferLayoutConstraint()
88 : LayoutConstraint(mandatory, dfs, priority), in OperandLayoutConstraint()
227 auto priority = LayoutConstraint::kDefaultPriority; in SetOperandLayout()
1264 std::deque<const LayoutConstraint*> worklist; in PropagateConstraints()
1282 const LayoutConstraint* layout_constraint = worklist.front(); in PropagateConstraints()