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)