Searched refs:buffer_constraint (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_layout_assignment.cc | 328 const BufferLayoutConstraint& buffer_constraint, in PropagateBufferConstraint() argument
330 const LogicalBuffer& buf = buffer_constraint.buffer(); in PropagateBufferConstraint()
334 *shape_with_layout.mutable_layout() = buffer_constraint.layout(); in PropagateBufferConstraint()
364 return LayoutAssignment::PropagateBufferConstraint(buffer_constraint, in PropagateBufferConstraint()
|
| D | gpu_layout_assignment.h | 47 const BufferLayoutConstraint& buffer_constraint,
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | layout_assignment.cc | 1141 if (auto* buffer_constraint = in PropagateConstraints() local
1144 PropagateBufferConstraint(*buffer_constraint, constraints)); in PropagateConstraints()
1379 const BufferLayoutConstraint& buffer_constraint, in PropagateBufferConstraintToOperands() argument
1382 << buffer_constraint.ToString(); in PropagateBufferConstraintToOperands()
1383 const LogicalBuffer& buffer = buffer_constraint.buffer(); in PropagateBufferConstraintToOperands()
1400 LayoutUtil::MinorToMajor(buffer_constraint.layout()).size()) { in PropagateBufferConstraintToOperands()
1402 buffer_constraint.layout(), instruction, operand_no, in PropagateBufferConstraintToOperands()
1417 ChooseOperandLayoutFromOutputLayout(buffer_constraint.layout(), in PropagateBufferConstraintToOperands()
1434 const BufferLayoutConstraint& buffer_constraint, in PropagateBufferConstraint() argument
1437 const LogicalBuffer& buffer = buffer_constraint.buffer(); in PropagateBufferConstraint()
[all …]
|
| 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()
|
| D | layout_assignment.h | 416 const BufferLayoutConstraint& buffer_constraint,
|