code: def test_sparse_int64_2d(batch_size, benchmark):
      run_atds_benchmark(
          TensorType.SPARSE,
          2,
          tf.int64,
          SMALL_NUM_RECORDS,
          SINGLE_PARTITION,
          batch_size,
          benchmark,
      )
  
file_path: io/tests/test_atds_avro/benchmark/test_sparse_tensor_atds_benchmark.py
imports: import pytest
  import tensorflow
  from tests.test_atds_avro.utils.data_source_registry import TensorType, SMALL_NUM_RECORDS, SINGLE_PARTITION
  from tests.test_atds_avro.utils.atds_benchmark_utils import run_atds_benchmark
label_desc: Use image node_id 4 for calling a global function with example usage: test_sparse_int64_2d(batch_size, benchmark) without return types