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.h92 class OperandLayoutConstraint : public LayoutConstraint {
94 OperandLayoutConstraint(const ShapeLayout& shape_layout,
161 const OperandLayoutConstraint* GetOperandLayoutConstraint(
215 std::map<OperandConstraintKey, OperandLayoutConstraint> operand_constraints_;
334 const OperandLayoutConstraint& layout_constraint,
Dlayout_assignment.cc77 OperandLayoutConstraint::OperandLayoutConstraint( in OperandLayoutConstraint() function in xla::OperandLayoutConstraint
92 string OperandLayoutConstraint::ToString() const { in ToString()
206 const OperandLayoutConstraint* curr_shape_layout = in SetOperandLayout()
238 key, OperandLayoutConstraint(ShapeLayout(shape_with_layout), in SetOperandLayout()
243 OperandLayoutConstraint(ShapeLayout(shape_with_layout), instruction, in SetOperandLayout()
344 const OperandLayoutConstraint* LayoutConstraints::GetOperandLayoutConstraint( in GetOperandLayoutConstraint()
1146 dynamic_cast<const OperandLayoutConstraint*>( in PropagateConstraints()
1237 const OperandLayoutConstraint& operand_constraint, in PropagateOperandConstraint()
1292 const OperandLayoutConstraint* constraint = in PropagateOperandConstraint()
1413 const OperandLayoutConstraint* constraint = in PropagateBufferConstraintToOperands()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment.h44 const OperandLayoutConstraint& layout_constraint,
Dgpu_layout_assignment.cc275 const OperandLayoutConstraint& layout_constraint, in PropagateOperandConstraint()