Home
last modified time | relevance | path

Searched refs:buffer_constraint (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dlayout_assignment.cc479 auto* buffer_constraint = GetBufferLayoutConstraint(*buffer); in ToString() local
480 if (buffer_constraint != nullptr) { in ToString()
482 LayoutUtil::HumanString(buffer_constraint->layout()), in ToString()
1431 if (auto* buffer_constraint = in PropagateConstraints() local
1434 PropagateBufferConstraint(*buffer_constraint, constraints)); in PropagateConstraints()
1694 auto* buffer_constraint = GetBufferLayoutConstraint(*buffer); in PropagateOperandConstraint() local
1695 if (buffer_constraint == nullptr || !buffer_constraint->mandatory()) { in PropagateOperandConstraint()
1713 const BufferLayoutConstraint& buffer_constraint, in PropagateBufferConstraintToOperands() argument
1716 << buffer_constraint.ToString(); in PropagateBufferConstraintToOperands()
1717 const LogicalBuffer& buffer = buffer_constraint.buffer(); in PropagateBufferConstraintToOperands()
[all …]
H A Dlayout_assignment.h398 const BufferLayoutConstraint& buffer_constraint,
437 const BufferLayoutConstraint& buffer_constraint,
521 const BufferLayoutConstraint& buffer_constraint,
H A Dlayout_assignment_test.cc485 const BufferLayoutConstraint& buffer_constraint, in PropagateBufferConstraint() argument
487 const LogicalBuffer& buffer = buffer_constraint.buffer(); in PropagateBufferConstraint()
497 TF_RETURN_IF_ERROR(SetArrayOperandLayout(buffer_constraint.layout(), in PropagateBufferConstraint()
501 return PropagateBufferConstraintToUses(buffer_constraint, constraints); in PropagateBufferConstraint()