.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")

@composite.Composite('TestIdentityNOp')
def _tfr_tensor_tensor_list(x):
  return x


@composite.Composite('TestInputNOp')
def _tfr_tensor_tensor_list_get_elt(x):
  return x[1]


@composite.Composite('TestOutputNOp')
def _tfr_tensor_tensor_list_output(x):
  return [x, x]


@composite.Composite('TestTwoInputsOp')
def _tfr_tensor_tensor_list_split(x, y, pred):
  z, _ = array_ops.Split(axis=0, value=x, num_split=2)