.Attr("N: int >= 1")
    .Attr("T: numbertype");

REGISTER_OP("TestInputNOp")
    .Input("input: N * T")
    .Output("output: T")
    .Attr("N: int >= 1")
    .Attr("T: numbertype");

REGISTER_OP("TestOutputNOp")
    .Input("input: T")
    .Output("output: N * T")
    .Attr("N: int >= 1")
    .Attr("T: numbertype");

REGISTER_OP("TestTwoInputsOp")
    .Input("lhs: T")
    .Input("rhs: T")
    .Output("output: T")