load("@org_xprof//plugin/xprof/build_utils:profiler_test.bzl", "profiler_test")
load("//plugin/xprof/build_utils:strict.default.bzl", "py_strict_library", "py_strict_test")

profiler_test(
    name = "tpu_tf2_keras_xla_test",
    srcs = ["tpu_tf2_keras_test.py"],
    main = "tpu_tf2_keras_test.py",
    platforms = [
        "v2",
        "v3",
        "v4",
    ],
    deps = [
        "//:expect_tensorflow_installed",
        "//plugin/xprof/convert:raw_to_tool_data",
        "//plugin/xprof/integration_tests:tf_mnist",
        "//plugin/xprof/integration_tests:tf_profiler_session",
        "@absl_py//absl/flags",
        "@absl_py//absl/logging",
        "@absl_py//absl/testing:absltest",
    ],
)

py_strict_library(
    name = "tpu_tf2_keras_xla",
    srcs = ["tpu_tf2_keras_test.py"],
    visibility = ["//plugin:__pkg__"],
    deps = [
        "//:expect_tensorflow_installed",
        "//plugin/xprof/convert:raw_to_tool_data",
        "//plugin/xprof/integration_tests:tf_mnist",
        "//plugin/xprof/integration_tests:tf_profiler_session",
        "@absl_py//absl/flags",
        "@absl_py//absl/logging",
        "@absl_py//absl/testing:absltest",
    ],
)
