def METHOD_NAME(batch_size, benchmark):
    run_atds_benchmark(
        TensorType.DENSE,
        0,
        tf.float64,
        SMALL_NUM_RECORDS,
        SINGLE_PARTITION,
        batch_size,
        benchmark,
    )