],
    hdrs = [
        "utils/size_utils.h",
    ],
    deps = [
        "@llvm-project//mlir:IR",
    ],
)

tf_cc_test(
    name = "size_utils_test",
    size = "small",
    srcs = ["utils/size_utils_test.cc"],
    deps = [
        ":size_utils",
        "//tensorflow/core:test",
        "//tensorflow/core:test_main",
        "@llvm-project//mlir:IR",
    ],
)