Home
last modified time | relevance | path

Searched refs:BufferLayoutConstraint (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment.h74 class BufferLayoutConstraint : public LayoutConstraint {
76 BufferLayoutConstraint(const Layout& layout, const LogicalBuffer& buffer,
159 const BufferLayoutConstraint* GetBufferLayoutConstraint(
212 std::unordered_map<const LogicalBuffer*, BufferLayoutConstraint>
333 const BufferLayoutConstraint& layout_constraint,
353 const BufferLayoutConstraint& layout_constraint,
424 const BufferLayoutConstraint& buffer_constraint,
Dlayout_assignment.cc70 BufferLayoutConstraint::BufferLayoutConstraint(const Layout& layout, in BufferLayoutConstraint() function in xla::BufferLayoutConstraint
77 string BufferLayoutConstraint::ToString() const { in ToString()
177 const BufferLayoutConstraint& curr_constraint = iter->second; in SetBufferLayout()
194 iter->second = BufferLayoutConstraint(layout, buffer, mandatory, dfs); in SetBufferLayout()
201 BufferLayoutConstraint(layout, buffer, mandatory, dfs))) in SetBufferLayout()
339 const BufferLayoutConstraint* LayoutConstraints::GetBufferLayoutConstraint( in GetBufferLayoutConstraint()
1200 dynamic_cast<const BufferLayoutConstraint*>(layout_constraint)) { in PropagateConstraints()
1334 const BufferLayoutConstraint* constraint = in PropagateOperandConstraint()
1434 const BufferLayoutConstraint* constraint = in PropagateOperandConstraint()
1484 const BufferLayoutConstraint& buffer_constraint, in PropagateBufferConstraintToOperands()
[all …]
Dlayout_assignment_test.cc522 const BufferLayoutConstraint& buffer_constraint, in PropagateBufferConstraint()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment.h47 const BufferLayoutConstraint& buffer_constraint,
Dgpu_layout_assignment.cc361 const BufferLayoutConstraint& buffer_constraint, in PropagateBufferConstraint()