import tensorflow as tf

indices = tf.constant(53, shape=[3], dtype=tf.int64)
values = tf.constant(0.554979503, shape=[218650], dtype=tf.float32)
dense shape = tf.constant(53, shape=[3], dtype=tf.int64)
  
tf.raw ops.SparseTensorToCSRSparseMatrix(
 indices=indices,
 values=values,
 dense shape=dense shape)

  const Tensor& indices = ctx->input(0);
  const Tensor& values = ctx->input(1);
  const Tensor& dense shape = ctx->input(2);
  const int rank = dense shape.NumElements();
  OP REQUIRES(ctx, rank == 2 || rank == 3,
        errors::InvalidArgument("SparseTensor must have rank 2 or 3; ",
                    "but indices has rank: ", rank));
  auto dense shape vec = dense shape.vec<int64 t>();
  // ...
  OP REQUIRES OK(
    ctx,
    coo to csr(batch size, num rows, indices.template matrix<int64 t>(),
          batch ptr.vec<int32>(), csr row ptr.vec<int32>(),
          csr col ind.vec<int32>()));