Ejemplo n.º 1
0
def agg_next_fn(state, value, weight):
    @tff.tf_computation(tf.int32)
    def add_one(value):
        return value + 1

    return {
        'call_count': tff.federated_apply(add_one, state.call_count),
    }, tff.federated_mean(value, weight)
Ejemplo n.º 2
0
 def _(x, y):
   return tff.federated_mean(x, y)
Ejemplo n.º 3
0
 def foo(x, y):
   return tff.federated_mean(x, y)