def _(x):
return intrinsics.federated_collect(x)
def foo(x):
val = intrinsics.federated_collect(x)
self.assertIsInstance(val, value_base.Value)
return val
def simple_collect():
one_at_clients = intrinsics.federated_value(1, placement_literals.CLIENTS)
return intrinsics.federated_collect(one_at_clients)
def trivial_collect():
empty_at_clients = intrinsics.federated_value((),
placement_literals.CLIENTS)
return intrinsics.federated_collect(empty_at_clients)
def bar():
x = intrinsics.federated_value(make_dataset(), placements.CLIENTS)
return intrinsics.federated_map(
foo,
intrinsics.federated_collect(intrinsics.federated_map(foo, x)))
def comp():
x = intrinsics.federated_value(10, placements.CLIENTS)
return intrinsics.federated_collect(x)
