code: def parse_args():
      """Function to parse arguments"""
      parser = argparse.ArgumentParser()
      parser.add_argument(
          "--lhs-row-dim", required=True, help="Provide batch_size"
      )
      parser.add_argument(
          "--lhs-col-dim", required=True, help="Provide feature_dim"
      )
      parser.add_argument(
          "--rhs-col-dim", required=True, help="Provide output_dim"
      )
      parser.add_argument(
          "--density", required=True, help="Density for lhs"
      )
      parser.add_argument(
          "--num-omp-threads",
          type=int,
          default=1,
          help="number of omp threads to set in MXNet",
      )
      parser.add_argument(
          "--lhs-stype",
          default="csr",
          choices=["csr", "default", "row_sparse"],
          help="stype for lhs",
          required=True,
      )
      parser.add_argument(
          "--rhs-stype",
          default="default",
          choices=["default", "row_sparse"],
          help="rhs stype",
          required=True,
      )
      parser.add_argument(
          "--only-storage", action="store_true", help="only storage"
      )
      parser.add_argument("--rhs-density", help="rhs_density")
      return parser.parse_args()
  
file_path: mxnet/benchmark/python/sparse/memory_benchmark.py
imports: import ctypes
  import sys
  import argparse
  import mxnet
  from mxnet.test_utils import rand_ndarray
  from mxnet.base import check_call, _LIB
label_desc: Use image node_id 1 for calling a global function with example usage: parse_args() and returns: parser