131 lines
4.2 KiB
Text
131 lines
4.2 KiB
Text
load("@rules_python//python:defs.bzl", "py_binary", "py_library")
|
|
|
|
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",
|
|
"//third_party/py/tensorflow:tensorflow_estimator",
|
|
],
|
|
)
|
|
|
|
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",
|
|
"@com_google_differential_py//python/dp_accounting:dp_event",
|
|
"@com_google_differential_py//python/dp_accounting/rdp:rdp_privacy_accountant",
|
|
],
|
|
)
|
|
|
|
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",
|
|
"@com_google_differential_py//python/dp_accounting:dp_event",
|
|
"@com_google_differential_py//python/dp_accounting/rdp:rdp_privacy_accountant",
|
|
],
|
|
)
|
|
|
|
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",
|
|
"@com_google_differential_py//python/dp_accounting:dp_event",
|
|
"@com_google_differential_py//python/dp_accounting/rdp:rdp_privacy_accountant",
|
|
],
|
|
)
|
|
|
|
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",
|
|
"//third_party/py/tensorflow:tensorflow_compat_v1_estimator",
|
|
"//third_party/py/tensorflow:tensorflow_estimator",
|
|
"@com_google_differential_py//python/dp_accounting:dp_event",
|
|
"@com_google_differential_py//python/dp_accounting/rdp:rdp_privacy_accountant",
|
|
],
|
|
)
|
|
|
|
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",
|
|
"//third_party/py/tensorflow:tensorflow_estimator",
|
|
],
|
|
)
|
|
|
|
py_binary(
|
|
name = "mnist_lr_tutorial",
|
|
srcs = ["mnist_lr_tutorial.py"],
|
|
python_version = "PY3",
|
|
srcs_version = "PY3",
|
|
deps = [
|
|
"//tensorflow_privacy/privacy/analysis:rdp_accountant",
|
|
"//tensorflow_privacy/privacy/optimizers:dp_optimizer",
|
|
"//third_party/py/tensorflow:tensorflow_compat_v1_estimator",
|
|
"//third_party/py/tensorflow:tensorflow_estimator",
|
|
"@com_google_differential_py//python/dp_accounting:dp_event",
|
|
"@com_google_differential_py//python/dp_accounting/rdp:rdp_privacy_accountant",
|
|
],
|
|
)
|
|
|
|
py_binary(
|
|
name = "lm_dpsgd_tutorial",
|
|
srcs = ["lm_dpsgd_tutorial.py"],
|
|
python_version = "PY3",
|
|
srcs_version = "PY3",
|
|
deps = [
|
|
"//tensorflow_privacy/privacy/optimizers:dp_optimizer",
|
|
"//third_party/py/tensorflow:tensorflow_compat_v1_estimator",
|
|
"//third_party/py/tensorflow:tensorflow_estimator",
|
|
"@com_google_differential_py//python/dp_accounting:dp_event",
|
|
"@com_google_differential_py//python/dp_accounting/rdp:rdp_privacy_accountant",
|
|
],
|
|
)
|
|
|
|
py_binary(
|
|
name = "movielens_tutorial",
|
|
srcs = ["movielens_tutorial.py"],
|
|
deps = [
|
|
"//tensorflow_privacy/privacy/analysis:gdp_accountant",
|
|
"//tensorflow_privacy/privacy/optimizers:dp_optimizer",
|
|
"//third_party/py/tensorflow:tensorflow_compat_v1_estimator",
|
|
"//third_party/py/tensorflow:tensorflow_estimator",
|
|
],
|
|
)
|
|
|
|
filegroup(
|
|
name = "ignore_srcs",
|
|
srcs = ["bolton_tutorial.py"],
|
|
tags = ["ignore_srcs"],
|
|
)
|