forked from 626_privacy/tensorflow_privacy
56 lines
1,002 B
Text
56 lines
1,002 B
Text
|
licenses(["notice"]) # Apache 2.0
|
||
|
|
||
|
py_library(
|
||
|
name = "gaussian_average_query",
|
||
|
srcs = ["gaussian_average_query.py"],
|
||
|
deps = [
|
||
|
":private_queries",
|
||
|
"//third_party/py/tensorflow",
|
||
|
],
|
||
|
)
|
||
|
|
||
|
py_library(
|
||
|
name = "dp_optimizers",
|
||
|
deps = [
|
||
|
":dp_adam",
|
||
|
":dp_gradient_descent",
|
||
|
],
|
||
|
)
|
||
|
|
||
|
py_library(
|
||
|
name = "dp_adam",
|
||
|
srcs = [
|
||
|
"dp_adam.py",
|
||
|
],
|
||
|
deps = [
|
||
|
":gaussian_average_query",
|
||
|
"//third_party/py/tensorflow",
|
||
|
],
|
||
|
)
|
||
|
|
||
|
py_library(
|
||
|
name = "dp_gradient_descent",
|
||
|
srcs = [
|
||
|
"dp_gradient_descent.py",
|
||
|
],
|
||
|
deps = [
|
||
|
":gaussian_average_query",
|
||
|
"//third_party/py/tensorflow",
|
||
|
],
|
||
|
)
|
||
|
|
||
|
py_test(
|
||
|
name = "dp_optimizer_test",
|
||
|
srcs = ["dp_optimizer_test.py"],
|
||
|
deps = [
|
||
|
":dp_optimizers",
|
||
|
"//third_party/py/absl/testing:parameterized",
|
||
|
"//third_party/py/tensorflow",
|
||
|
],
|
||
|
)
|
||
|
|
||
|
py_library(
|
||
|
name = "private_queries",
|
||
|
srcs = ["private_queries.py"],
|
||
|
)
|