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.h72 class BufferLayoutConstraint : public LayoutConstraint {
74 BufferLayoutConstraint(const Layout& layout, const LogicalBuffer& buffer,
157 const BufferLayoutConstraint* GetBufferLayoutConstraint(
210 std::unordered_map<const LogicalBuffer*, BufferLayoutConstraint>
331 const BufferLayoutConstraint& layout_constraint,
348 const BufferLayoutConstraint& layout_constraint,
416 const BufferLayoutConstraint& buffer_constraint,
Dlayout_assignment.cc65 BufferLayoutConstraint::BufferLayoutConstraint(const Layout& layout, in BufferLayoutConstraint() function in xla::BufferLayoutConstraint
72 string BufferLayoutConstraint::ToString() const { in ToString()
172 const BufferLayoutConstraint& curr_constraint = iter->second; in SetBufferLayout()
184 iter->second = BufferLayoutConstraint(layout, buffer, mandatory, dfs); in SetBufferLayout()
191 BufferLayoutConstraint(layout, buffer, mandatory, dfs))) in SetBufferLayout()
329 const BufferLayoutConstraint* LayoutConstraints::GetBufferLayoutConstraint( in GetBufferLayoutConstraint()
1142 dynamic_cast<const BufferLayoutConstraint*>(layout_constraint)) { in PropagateConstraints()
1333 const BufferLayoutConstraint* constraint = in PropagateOperandConstraint()
1379 const BufferLayoutConstraint& buffer_constraint, in PropagateBufferConstraintToOperands()
1434 const BufferLayoutConstraint& buffer_constraint, in PropagateBufferConstraint()
[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.cc328 const BufferLayoutConstraint& buffer_constraint, in PropagateBufferConstraint()