def foo(x):
plus = computations.tf_computation(tf.add)
val = intrinsics.federated_reduce(x, 0, plus)
self.assertIsInstance(val, value_base.Value)
return val
def foo(x):
plus = computations.tf_computation(lambda a, b: tf.add(a, b)) # pylint: disable=unnecessary-lambda
val = intrinsics.federated_reduce(x, 0, plus)
self.assertIsInstance(val, value_base.Value)
return val
def bar(x): return intrinsics.federated_reduce(x, 0, foo)
def simple_reduce():
one_at_clients = intrinsics.federated_value(1, placement_literals.CLIENTS)
zero = 0
op = computations.tf_computation(lambda a, b: a + b)
return intrinsics.federated_reduce(one_at_clients, zero, op)
def trivial_reduce():
empty_at_clients = intrinsics.federated_value((),
placement_literals.CLIENTS)
zero = ()
op = computations.tf_computation(lambda _a, _b: ())
return intrinsics.federated_reduce(empty_at_clients, zero, op)
def comp():
return intrinsics.federated_reduce(
intrinsics.federated_value(10, placements.CLIENTS), 0,
add_numbers)
def foo(x):
plus = computations.tf_computation(tf.add, [tf.int32, tf.int32])
return intrinsics.federated_reduce(x, 0, plus)
def foo(x):
return intrinsics.federated_reduce(x, 0., add)
