Home
last modified time | relevance | path

Searched refs:layout_constraint (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment.cc275 const OperandLayoutConstraint& layout_constraint, in PropagateOperandConstraint() argument
277 const HloInstruction* instruction = layout_constraint.instruction(); in PropagateOperandConstraint()
284 layout_constraint.operand_no() == 0) { in PropagateOperandConstraint()
286 layout_constraint.shape_layout().shape(), instruction)); in PropagateOperandConstraint()
296 layout_constraint.operand_no() == 0) { in PropagateOperandConstraint()
301 layout_constraint.shape_layout().layout(), *out_buf)); in PropagateOperandConstraint()
310 (layout_constraint.operand_no() == 0 || in PropagateOperandConstraint()
311 layout_constraint.operand_no() == 4)) { in PropagateOperandConstraint()
316 layout_constraint.shape_layout().layout(), *out_buf)); in PropagateOperandConstraint()
318 int64 operand_to_set = layout_constraint.operand_no() == 0 ? 4 : 0; in PropagateOperandConstraint()
[all …]
Dgpu_layout_assignment.h44 const OperandLayoutConstraint& layout_constraint,
/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment.h331 const BufferLayoutConstraint& layout_constraint,
334 const OperandLayoutConstraint& layout_constraint,
337 const ResultLayoutConstraint& layout_constraint,
348 const BufferLayoutConstraint& layout_constraint,
Dlayout_assignment.cc1137 const LayoutConstraint* layout_constraint = worklist.front(); in PropagateConstraints() local
1139 VLOG(2) << "Propagating " << layout_constraint->ToString() in PropagateConstraints()
1142 dynamic_cast<const BufferLayoutConstraint*>(layout_constraint)) { in PropagateConstraints()
1147 layout_constraint)) { in PropagateConstraints()
1152 layout_constraint)) { in PropagateConstraints()
1156 LOG(FATAL) << "Invalid constraint type: " << *layout_constraint; in PropagateConstraints()
1471 const ResultLayoutConstraint& layout_constraint, in PropagateResultConstraint() argument
1476 layout_constraint.shape_layout(), in PropagateResultConstraint()