tensorflow_privacy/tutorials/BUILD
Richard Levasseur 6248be8290 Internal Code Change
PiperOrigin-RevId: 557198768
2023-08-15 11:42:02 -07:00

94 lines
2.5 KiB
Text

licenses(["notice"])
py_library(
name = "mnist_dpsgd_tutorial_common",
srcs = ["mnist_dpsgd_tutorial_common.py"],
srcs_version = "PY3",
)
py_binary(
name = "mnist_dpsgd_tutorial",
srcs = ["mnist_dpsgd_tutorial.py"],
python_version = "PY3",
srcs_version = "PY3",
deps = [
":mnist_dpsgd_tutorial_common",
"//tensorflow_privacy/privacy/analysis:compute_dp_sgd_privacy_lib",
"//tensorflow_privacy/privacy/optimizers:dp_optimizer",
],
)
py_binary(
name = "mnist_dpsgd_tutorial_eager",
srcs = ["mnist_dpsgd_tutorial_eager.py"],
python_version = "PY3",
srcs_version = "PY3",
deps = ["//tensorflow_privacy/privacy/optimizers:dp_optimizer"],
)
py_binary(
name = "mnist_dpsgd_tutorial_keras",
srcs = ["mnist_dpsgd_tutorial_keras.py"],
python_version = "PY3",
srcs_version = "PY3",
deps = ["//tensorflow_privacy/privacy/optimizers:dp_optimizer_keras"],
)
py_binary(
name = "mnist_dpsgd_tutorial_keras_model",
srcs = ["mnist_dpsgd_tutorial_keras_model.py"],
python_version = "PY3",
srcs_version = "PY3",
deps = ["//tensorflow_privacy/privacy/keras_models:dp_keras_model"],
)
py_binary(
name = "mnist_dpsgd_tutorial_vectorized",
srcs = ["mnist_dpsgd_tutorial_vectorized.py"],
python_version = "PY3",
srcs_version = "PY3",
deps = ["//tensorflow_privacy/privacy/optimizers:dp_optimizer_vectorized"],
)
py_binary(
name = "mnist_dpsgd_tutorial_tpu",
srcs = ["mnist_dpsgd_tutorial_tpu.py"],
python_version = "PY3",
srcs_version = "PY3",
deps = [
":mnist_dpsgd_tutorial_common",
"//tensorflow_privacy/privacy/analysis:compute_dp_sgd_privacy_lib",
"//tensorflow_privacy/privacy/optimizers:dp_optimizer",
],
)
py_binary(
name = "mnist_lr_tutorial",
srcs = ["mnist_lr_tutorial.py"],
python_version = "PY3",
srcs_version = "PY3",
deps = ["//tensorflow_privacy/privacy/optimizers:dp_optimizer"],
)
py_binary(
name = "lm_dpsgd_tutorial",
srcs = ["lm_dpsgd_tutorial.py"],
python_version = "PY3",
srcs_version = "PY3",
deps = ["//tensorflow_privacy/privacy/optimizers:dp_optimizer"],
)
py_binary(
name = "movielens_tutorial",
srcs = ["movielens_tutorial.py"],
deps = [
"//tensorflow_privacy/privacy/analysis:gdp_accountant",
"//tensorflow_privacy/privacy/optimizers:dp_optimizer",
],
)
filegroup(
name = "ignore_srcs",
srcs = ["bolton_tutorial.py"],
tags = ["ignore_srcs"],
)