let hasCustomAssemblyFormat = 0;
}

def TfExecutor_SwitchOp : TfExecutor_Op<"Switch",
    [ControlOperandsAfterAllData, HasParent<"GraphOp">,
     PredOpTrait<"data operand must be broadcastable to true result",
                 TF_OpIsBroadcastableToRes<0, 0>>,
     PredOpTrait<"data operand must be broadcastable to false result",
                 TF_OpIsBroadcastableToRes<0, 1>>]>{
  let summary = [{

class TF_OperandHasRank<int n, int m> :
  PredOpTrait<"operand " # n # " is " # m # "-D",
    Or<[TF_OperandIsUnrankedPred<n>,
      CPred<"$_op.getOperand(" # n #
      ").getType().cast<ShapedType>().getRank() == " # m>]>>;

// Returns true if the n-th result has unknown rank or has rank m.
class TF_ResultHasRank<int n, int m> :
  PredOpTrait<"result " # n # " is " # m # "-D",
    Or<[TF_ResultIsUnrankedPred<n>,