PredOpTrait<"operand " # x # " has num of elements equals to dim " # j #
    " of operand " # y,
    TFL_NumElementsEqualsDim<x, y, j>>;

// Return true if number of elements of x-th operand equals to n.
class TFL_NumElements<int x, int n> :
  Or<[TFL_OperandIsUnrankedPred<x>,
      CPred<"!$_op.getOperand(" # x #
        ").getType().cast<ShapedType>().hasStaticShape()">,
      CPred<"$_op.getOperand(" # x #