@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)