Home
last modified time | relevance | path

Searched refs:OperandLayout (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_layout_assignment.cc132 if (constraints->OperandLayout(instruction, operand_no) != nullptr) { in AddBackendConstraints()
/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment.cc335 const ShapeLayout* LayoutConstraints::OperandLayout( in OperandLayout() function in xla::LayoutConstraints
361 if (OperandLayout(instruction, i) != nullptr) { in ToString()
363 "): ", OperandLayout(instruction, i)->ToString(), "\n"); in ToString()
1460 if (constraints->OperandLayout(user, operand_no) == nullptr && in PropagateBufferConstraintToUses()
1597 constraints.OperandLayout(instruction, operand_no); in AssignLayouts()
Dlayout_assignment.h159 const ShapeLayout* OperandLayout(const HloInstruction* instruction,