#     # copybara:uncomment driver = "//tensorflow/compiler/mlir:run_lit.sh",
#     exclude = [
#         "compile.benchmark.large.mlir",
#         "batch_function_fallback.mlir",
#         "create_op.mlir",
#         "custom_thread_pool.mlir",
#     ],
#     # copybara:uncomment flaky = ["compile.error.mlir"],
#     size_override = {
#         "compile.benchmark.small.mlir": "medium",
#         "batching_fallback.mlir": "medium",
#     },