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.cc308 const OperandLayoutConstraint& layout_constraint, in PropagateOperandConstraint() argument
310 const HloInstruction* instruction = layout_constraint.instruction(); in PropagateOperandConstraint()
317 layout_constraint.operand_no() == 0) { in PropagateOperandConstraint()
319 layout_constraint.shape_layout().shape(), instruction)); in PropagateOperandConstraint()
329 layout_constraint.operand_no() == 0) { in PropagateOperandConstraint()
334 layout_constraint.shape_layout().layout(), *out_buf)); in PropagateOperandConstraint()
343 (layout_constraint.operand_no() == 0 || in PropagateOperandConstraint()
344 layout_constraint.operand_no() == 4)) { in PropagateOperandConstraint()
349 layout_constraint.shape_layout().layout(), *out_buf)); in PropagateOperandConstraint()
351 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.h333 const BufferLayoutConstraint& layout_constraint,
336 const OperandLayoutConstraint& layout_constraint,
339 const ResultLayoutConstraint& layout_constraint,
353 const BufferLayoutConstraint& layout_constraint,
Dlayout_assignment.cc1195 const LayoutConstraint* layout_constraint = worklist.front(); in PropagateConstraints() local
1197 VLOG(2) << "Propagating " << layout_constraint->ToString() in PropagateConstraints()
1200 dynamic_cast<const BufferLayoutConstraint*>(layout_constraint)) { in PropagateConstraints()
1205 layout_constraint)) { in PropagateConstraints()
1210 layout_constraint)) { in PropagateConstraints()
1214 LOG(FATAL) << "Invalid constraint type: " << *layout_constraint; in PropagateConstraints()
1612 const ResultLayoutConstraint& layout_constraint, in PropagateResultConstraint() argument
1617 layout_constraint.shape_layout(), in PropagateResultConstraint()