Searched refs:SetArrayOperandLayout (Results 1 – 3 of 3) sorted by relevance
261 Status LayoutConstraints::SetArrayOperandLayout( in SetArrayOperandLayout() function in xla::LayoutConstraints1404 TF_RETURN_IF_ERROR(constraints->SetArrayOperandLayout( in PropagateOperandConstraint()1496 TF_RETURN_IF_ERROR(constraints->SetArrayOperandLayout( in PropagateBufferConstraintToOperands()1513 TF_RETURN_IF_ERROR(constraints->SetArrayOperandLayout( in PropagateBufferConstraintToOperands()1532 TF_RETURN_IF_ERROR(constraints->SetArrayOperandLayout( in PropagateBufferConstraintToOperands()1576 TF_RETURN_IF_ERROR(constraints->SetArrayOperandLayout( in PropagateBufferConstraintToUses()
179 Status SetArrayOperandLayout(const Layout& layout,
534 TF_RETURN_IF_ERROR(constraints->SetArrayOperandLayout( in PropagateBufferConstraint()