Searched refs:buffer_constraint (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | layout_assignment.cc | 1287 if (auto* buffer_constraint = in PropagateConstraints() local
1290 PropagateBufferConstraint(*buffer_constraint, constraints)); in PropagateConstraints()
1572 const BufferLayoutConstraint& buffer_constraint, in PropagateBufferConstraintToOperands() argument
1575 << buffer_constraint.ToString(); in PropagateBufferConstraintToOperands()
1576 const LogicalBuffer& buffer = buffer_constraint.buffer(); in PropagateBufferConstraintToOperands()
1585 buffer_constraint.layout(), instruction, in PropagateBufferConstraintToOperands()
1600 LayoutUtil::MinorToMajor(buffer_constraint.layout()).size()) { in PropagateBufferConstraintToOperands()
1602 buffer_constraint.layout(), instruction, operand_no, in PropagateBufferConstraintToOperands()
1617 ChooseOperandLayoutFromOutputLayout(buffer_constraint.layout(), in PropagateBufferConstraintToOperands()
1636 const BufferLayoutConstraint& buffer_constraint, in PropagateBufferConstraint() argument
[all …]
|
| D | layout_assignment.h | 347 const BufferLayoutConstraint& buffer_constraint,
367 const BufferLayoutConstraint& buffer_constraint,
443 const BufferLayoutConstraint& buffer_constraint,
|
| D | layout_assignment_test.cc | 522 const BufferLayoutConstraint& buffer_constraint, in PropagateBufferConstraint() argument
524 const LogicalBuffer& buffer = buffer_constraint.buffer(); in PropagateBufferConstraint()
535 buffer_constraint.layout(), instruction, operand_no, in PropagateBufferConstraint()
538 return PropagateBufferConstraintToUses(buffer_constraint, constraints); in PropagateBufferConstraint()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_layout_assignment.cc | 403 const BufferLayoutConstraint& buffer_constraint, in PropagateBufferConstraint() argument
405 const LogicalBuffer& buf = buffer_constraint.buffer(); in PropagateBufferConstraint()
409 *shape_with_layout.mutable_layout() = buffer_constraint.layout(); in PropagateBufferConstraint()
439 return LayoutAssignment::PropagateBufferConstraint(buffer_constraint, in PropagateBufferConstraint()
|
| D | gpu_layout_assignment.h | 47 const BufferLayoutConstraint& buffer_constraint,
|