Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dlayout_assignment.h333 Status SetArrayOperandLayout(const Layout& layout,
337 return SetArrayOperandLayout(layout, instruction, operand_no, mandatory,
340 Status SetArrayOperandLayout(const Layout& layout,
H A Dlayout_assignment.cc330 Status LayoutAssignment::SetArrayOperandLayout( in SetArrayOperandLayout() function in xla::LayoutAssignment
1639 TF_RETURN_IF_ERROR(SetArrayOperandLayout( in PropagateOperandConstraint()
1725 TF_RETURN_IF_ERROR(SetArrayOperandLayout( in PropagateBufferConstraintToOperands()
1742 TF_RETURN_IF_ERROR(SetArrayOperandLayout( in PropagateBufferConstraintToOperands()
1749 TF_RETURN_IF_ERROR(SetArrayOperandLayout( in PropagateBufferConstraintToOperands()
1769 TF_RETURN_IF_ERROR(SetArrayOperandLayout( in PropagateBufferConstraintToOperands()
1815 TF_RETURN_IF_ERROR(SetArrayOperandLayout( in PropagateBufferConstraintToUses()
H A Dlayout_assignment_test.cc497 TF_RETURN_IF_ERROR(SetArrayOperandLayout(buffer_constraint.layout(), in PropagateBufferConstraint()