code: def init_kv():
      # # init kv dns keys
      kv.init(keys_shape, [mx.nd.ones(shape)] * len(keys_shape))
      kv.init(
          keys_big_shape, [mx.nd.ones(big_shape)] * len(keys_big_shape)
      )
      # # init kv row_sparse keys
      kv.init(
          rsp_keys_shape,
          [mx.nd.ones(shape).tostype("row_sparse")]
          * len(rsp_keys_shape),
      )
      kv.init(
          rsp_keys_big_shape,
          [mx.nd.ones(big_shape).tostype("row_sparse")]
          * len(rsp_keys_big_shape),
      )
      # init fp16 dns keys
      kv.init(
          fp16_keys_shape,
          [mx.nd.ones(shape, dtype="float16")] * len(keys_shape),
      )
      kv.init(
          fp16_keys_big_shape,
          [mx.nd.ones(big_shape, dtype="float16")]
          * len(keys_big_shape),
      )
      # init fp16 row_sparse keys
      kv.init(
          fp16_rsp_keys_shape,
          [mx.nd.ones(shape, dtype="float16").tostype("row_sparse")]
          * len(fp16_rsp_keys_shape),
      )
      kv.init(
          fp16_rsp_keys_big_shape,
          [mx.nd.ones(big_shape, dtype="float16").tostype("row_sparse")]
          * len(fp16_rsp_keys_big_shape),
      )
      return kv
  
file_path: mxnet/tests/nightly/dist_sync_kvstore.py
imports: import sys
  import argparse
  import mxnet
  import numpy
  import numpy.random
  from mxnet.test_utils import assert_almost_equal, assert_exception
  from test_kvstore import compute_expected_quantization, compute_1bit, compute_2bit
label_desc: Use image node_id 2 for calling a global function with example usage: init_kv() and returns: kv