code: def benchmark_func(dataset):
      count = 0
      for _ in dataset:
          count += 1
      return count
  
file_path: io/tests/test_atds_avro/utils/benchmark_utils.py
imports: import os
  import tensorflow
  from tests.test_atds_avro.utils.data_source_registry import get_canonical_name, get_data_source_from_registry
  from tests.test_atds_avro.utils.generator.tensor_generator import IntTensorGenerator, FloatTensorGenerator, BoolTensorGenerator
  from tests.test_atds_avro.utils.generator.sparse_tensor_generator import IntSparseTensorGenerator, ValueDistribution
  from tests.test_atds_avro.utils.generator.varlen_tensor_generator import WordVarLenTensorGenerator, DimensionDistribution
  from tests.test_atds_avro.utils.tf_record_writer import TFRecordWriter
label_desc: Use image node_id 1 for calling a global function with example usage: benchmark_func(dataset) and returns: count