Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment.h94 class OperandLayoutConstraint : public LayoutConstraint {
96 OperandLayoutConstraint(const ShapeLayout& shape_layout,
163 const OperandLayoutConstraint* GetOperandLayoutConstraint(
217 std::map<OperandConstraintKey, OperandLayoutConstraint> operand_constraints_;
336 const OperandLayoutConstraint& layout_constraint,
Dlayout_assignment.cc82 OperandLayoutConstraint::OperandLayoutConstraint( in OperandLayoutConstraint() function in xla::OperandLayoutConstraint
97 string OperandLayoutConstraint::ToString() const { in ToString()
216 const OperandLayoutConstraint* curr_shape_layout = in SetOperandLayout()
248 key, OperandLayoutConstraint(ShapeLayout(shape_with_layout), in SetOperandLayout()
253 OperandLayoutConstraint(ShapeLayout(shape_with_layout), instruction, in SetOperandLayout()
354 const OperandLayoutConstraint* LayoutConstraints::GetOperandLayoutConstraint( in GetOperandLayoutConstraint()
1204 dynamic_cast<const OperandLayoutConstraint*>( in PropagateConstraints()
1303 const OperandLayoutConstraint& operand_constraint, in PropagateOperandConstraint()
1389 const OperandLayoutConstraint* constraint = in PropagateOperandConstraint()
1525 const OperandLayoutConstraint* constraint = in PropagateBufferConstraintToOperands()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment.h44 const OperandLayoutConstraint& layout_constraint,
Dgpu_layout_assignment.cc308 const OperandLayoutConstraint& layout_constraint, in PropagateOperandConstraint()