Skip to content

Commit 3fb7984

Browse files
sdieraufcopybara-github
authored andcommitted
Extend the demo server.py to take a custom aggregation protocol
PiperOrigin-RevId: 722615196
1 parent f8f1c86 commit 3fb7984

File tree

1 file changed

+2
-0
lines changed
  • tensorflow_federated/cc/core/impl/aggregation/tensorflow/python

1 file changed

+2
-0
lines changed

tensorflow_federated/cc/core/impl/aggregation/tensorflow/python/BUILD

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,8 @@ pybind_extension(
2626
deps = [
2727
"//tensorflow_federated/cc/core/impl/aggregation/protocol:aggregation_protocol",
2828
"//tensorflow_federated/cc/core/impl/aggregation/protocol:configuration_cc_proto",
29+
"//tensorflow_federated/cc/core/impl/aggregation/protocol:federated_compute_checkpoint_builder",
30+
"//tensorflow_federated/cc/core/impl/aggregation/protocol:federated_compute_checkpoint_parser",
2931
"//tensorflow_federated/cc/core/impl/aggregation/protocol:resource_resolver",
3032
"//tensorflow_federated/cc/core/impl/aggregation/protocol/simple_aggregation",
3133
"//tensorflow_federated/cc/core/impl/aggregation/tensorflow:tensorflow_checkpoint_builder_factory",

0 commit comments

Comments
 (0)