"@local_tsl//tsl/platform:test",
        "@local_tsl//tsl/platform:test_main",
    ],
)

tf_cuda_library(
    name = "checkpoint_reader",
    srcs = ["checkpoint_reader.cc"],
    hdrs = ["checkpoint_reader.h"],
    visibility = ["//visibility:public"],
    deps = [
        ":tf_status",
        ":tf_status_helper",
        "//tensorflow/core:framework",

    ],
)

filegroup(
    name = "headers",
    srcs = [
        "c_api.h",
        "c_api_experimental.h",
        "dlpack.h",
    ],
    visibility = ["//tensorflow:__subpackages__"],
)

filegroup(
    name = "pywrap_headers_monitoring_reader",
    srcs = [
        "c_api_experimental_reader.h",
        "tfe_monitoring_reader_internal.h",
    ],