tff.backends.mapreduce.get_state_initialization_computation_for_map_reduce_form

Stay organized with collections Save and categorize content based on your preferences.

Validates and transforms a computation to generate state for MapReduceForm.

initialize_computation A computation_base.Computation that should generate initial state for a computation that is compatible with MapReduceForm.
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.MapReduceForm. These options are combined with a set of defaults that aggressively configure Grappler. If the input grappler_config has graph_options.rewrite_options.disable_meta_optimizer=True, Grappler is bypassed.

A computation_base.Computation that can generate state for a computation that is compatible with MapReduceForm.

TypeError If the arguments are of the wrong types.