code: def create_test_table_shards(
      table: TestTable, num_sc_shards: int, table_data_start=0
  ):
      t = array_ops.reshape(
          math_ops.range(
              start=table_data_start,
              delta=1,
              limit=table_data_start + table.vocab * table.dim,
          ),
          (table.vocab, table.dim),
      )
      shards = [t[i::num_sc_shards, :] for i in range(num_sc_shards)]
      if table.shift:
          shards = collections.deque(shards)
          shards.rotate(table.shift)
          return (t, list(shards))
      else:
          return (t, shards)
  
file_path: tensorflow-upstream/tensorflow/python/tpu/tpu_embedding_v3_utils_test.py
imports: import collections
  from absl.testing import parameterized
  from tensorflow.python.compat import v2_compat
  from tensorflow.python.framework.constant_op import constant
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.platform import test
  from tensorflow.python.tpu import tpu_embedding_v3_utils
label_desc: Use image node_id 1 for calling a global function with example usage: create_test_table_shards(table, num_sc_shards, table_data_start) and returns: t,  list, t,  shards