tff.backends.mapreduce.get_distribute_aggregate_form_for_computation

Constructs DistributeAggregateForm for a computation.

tff.backends.mapreduce.get_distribute_aggregate_form_for_computation(
    comp: tff.framework.ConcreteComputation,
    *,
    tff_internal_preprocessing: Optional[BuildingBlockFn] = None
) -> tff.backends.mapreduce.DistributeAggregateForm

comp An instance of computation_impl.ConcreteComputation that is compatible with DistributeAggregateForm. The computation must take exactly two arguments, and the first must be a state value placed at SERVER. The computation must return exactly two values. The type of the first element in the result must also be assignable to the first element of the parameter.
tff_internal_preprocessing An optional function to transform the AST of the iterative process.

An instance of tff.backends.mapreduce.DistributeAggregateForm equivalent to the provided computation_base.Computation.

TypeError If the arguments are of the wrong types.