Constructs tff.backends.mapreduce.BroadcastForm given a computation.
tff.backends.mapreduce.get_broadcast_form_for_computation(
comp: tff.Computation,
grappler_config: Optional[tf.compat.v1.ConfigProto] = _GRAPPLER_DEFAULT_CONFIG
) -> tff.backends.mapreduce.BroadcastForm
Args |
|---|
comp
|
An instance of tff.Computation 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 optional 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
None, grappler is bypassed.
|
View source on GitHub