import tensorflow as tf

ckpt path = tf.constant(
  "/tmp/warm starting util test5kl2a3pc/tmpph76tep2/model-0", shape=[], dtype=tf.string)
old tensor name = tf.constant(
  "/tmp/warm starting util test5kl2a3pc/tmpph76tep2/model-0", shape=[], dtype=tf.string)

row remapping = tf.constant(0, shape=[], dtype=tf.int64)
col remapping = tf.constant(3, shape=[3], dtype=tf.int64)
initializing values = tf.constant([], shape=[0, 1], dtype=tf.float32)

tf.raw ops.LoadAndRemapMatrix(
 ckpt path=ckpt path,
 old tensor name=old tensor name,
 row remapping=row remapping,
 col remapping=col remapping,
 initializing values=initializing values,
 num rows=1,
 num cols=1)

OP REQUIRES OK(context, context->input("row remapping", &row remapping t));
const auto row remapping = row remapping t->vec<int64 t>();