Searched refs:buffer_constraint (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_layout_assignment.cc | 361 const BufferLayoutConstraint& buffer_constraint, in PropagateBufferConstraint() argument
363 const LogicalBuffer& buf = buffer_constraint.buffer(); in PropagateBufferConstraint()
367 *shape_with_layout.mutable_layout() = buffer_constraint.layout(); in PropagateBufferConstraint()
397 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 | 1199 if (auto* buffer_constraint = in PropagateConstraints() local
1202 PropagateBufferConstraint(*buffer_constraint, constraints)); in PropagateConstraints()
1484 const BufferLayoutConstraint& buffer_constraint, in PropagateBufferConstraintToOperands() argument
1487 << buffer_constraint.ToString(); in PropagateBufferConstraintToOperands()
1488 const LogicalBuffer& buffer = buffer_constraint.buffer(); in PropagateBufferConstraintToOperands()
1497 buffer_constraint.layout(), instruction, in PropagateBufferConstraintToOperands()
1512 LayoutUtil::MinorToMajor(buffer_constraint.layout()).size()) { in PropagateBufferConstraintToOperands()
1514 buffer_constraint.layout(), instruction, operand_no, in PropagateBufferConstraintToOperands()
1529 ChooseOperandLayoutFromOutputLayout(buffer_constraint.layout(), in PropagateBufferConstraintToOperands()
1548 const BufferLayoutConstraint& buffer_constraint, in PropagateBufferConstraint() argument
[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 | 424 const BufferLayoutConstraint& buffer_constraint,
|