code: def run_atds_benchmark(
      tensor_type,
      rank,
      dtype,
      num_records,
      partitions,
      batch_size,
      benchmark,
  ):
      data_source_name = get_canonical_name(
          tensor_type, rank, dtype, num_records, partitions
      )
      data_source = get_data_source_from_registry(data_source_name)
      run_atds_benchmark_from_data_source(
          data_source, batch_size, benchmark
      )
  
file_path: io/tests/test_atds_avro/utils/atds_benchmark_utils.py
imports: import avro.schema
  import glob
  import json
  import os
  import tensorflow
  from tests.test_atds_avro.utils.data_source import DataSource
  from tests.test_atds_avro.utils.data_source_registry import SMALL_NUM_RECORDS, get_canonical_name, get_data_source_from_registry
  from tests.test_atds_avro.utils.generator.tensor_generator import IntTensorGenerator, FloatTensorGenerator, WordTensorGenerator, BoolTensorGenerator
  from tests.test_atds_avro.utils.generator.sparse_tensor_generator import IntSparseTensorGenerator, FloatSparseTensorGenerator, WordSparseTensorGenerator, BoolSparseTensorGenerator, get_common_value_dist
  from tests.test_atds_avro.utils.generator.varlen_tensor_generator import IntVarLenTensorGenerator, FloatVarLenTensorGenerator, WordVarLenTensorGenerator, BoolVarLenTensorGenerator
  from tensorflow_io.python.experimental.atds.dataset import ATDSDataset
  from tests.test_atds_avro.utils.atds_writer import ATDSWriter
  from tests.test_atds_avro.utils.benchmark_utils import benchmark_func
label_desc: Use image node_id 4 for calling a global function with example usage: run_atds_benchmark(tensor_type, rank, dtype, num_records, partitions, batch_size, benchmark) without return types