tff.backends.mapreduce.get_broadcast_form_for_computation

Constructs tff.backends.mapreduce.BroadcastForm given a computation.

tff.backends.mapreduce.get_broadcast_form_for_computation(
    comp: tff.framework.ConcreteComputation,
    grappler_config: tf.compat.v1.ConfigProto = _GRAPPLER_DEFAULT_CONFIG,
    *,
    tff_internal_preprocessing: Optional[BuildingBlockFn] = None
) -> tff.backends.mapreduce.BroadcastForm

comp An instance of computation_impl.ConcreteComputation that is compatible with broadcast form. Computations are only compatible if they take in a single value placed at server, return a single value placed at clients, and do not contain any aggregations.
grappler_config An instance of tf.compat.v1.ConfigProto to configure Grappler graph optimization of the Tensorflow graphs backing the resulting tff.backends.mapreduce.BroadcastForm. These options are combined with a set of defaults that aggressively configure Grappler. If grappler_config_proto has graph_options.rewrite_options.disable_meta_optimizer=True, Grappler is bypassed.
tff_internal_preprocessing An optional function to transform the AST of the computation.

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