def _tfr_tf_ops_complex(lhs, rhs):
  left_padding, _ = array_ops.SplitV(
      value=lhs, size_splits=[rhs, -1], axis=0, num_split=2)
  _, right_padding = array_ops.SplitV(
      value=lhs, size_splits=[rhs, rhs], axis=1, num_split=2)
  return [left_padding, right_padding]


@composite.Composite('TestIdentityOp')
def _tfr_tf_ops_tensor(x):
  return array_ops.Identity(x)