tff.backends.mapreduce.get_state_initialization_computation
Stay organized with collections
Save and categorize content based on your preferences.
Validates and transforms a computation to generate state.
tff.backends.mapreduce.get_state_initialization_computation(
initialize_computation: tff.framework.ConcreteComputation,
grappler_config: tf.compat.v1.ConfigProto = _GRAPPLER_DEFAULT_CONFIG
) -> tff.Computation
Args |
|---|
initialize_computation
|
A computation_impl.ConcreteComputation that should
generate initial state for a computation that is compatible with a
federated learning system that implements the contract of a backend
defined in the backends/mapreduce directory.
|
grappler_config
|
An optional instance of tf.compat.v1.ConfigProto to
configure Grappler graph optimization of the TensorFlow graphs. 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.
|
Returns |
|---|
A computation_base.Computation that can generate state for a computation.
|
Raises |
|---|
TypeError
|
If the arguments are of the wrong types.
|
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.
Last updated 2024-09-20 UTC.
[null,null,["Last updated 2024-09-20 UTC."],[],[],null,["# tff.backends.mapreduce.get_state_initialization_computation\n\n\u003cbr /\u003e\n\n|-------------------------------------------------------------------------------------------------------------------------------------------------------------|\n| [View source on GitHub](https://github.com/tensorflow/federated/blob/v0.87.0 Version 2.0, January 2004 Licensed under the Apache License, Version 2.0 (the) |\n\nValidates and transforms a computation to generate state. \n\n tff.backends.mapreduce.get_state_initialization_computation(\n initialize_computation: ../../../tff/framework/ConcreteComputation,\n grappler_config: tf.compat.v1.ConfigProto = _GRAPPLER_DEFAULT_CONFIG\n ) -\u003e ../../../tff/Computation\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n| Args ---- ||\n|--------------------------|------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|\n| `initialize_computation` | A `computation_impl.ConcreteComputation` that should generate initial state for a computation that is compatible with a federated learning system that implements the contract of a backend defined in the backends/mapreduce directory. |\n| `grappler_config` | An optional instance of [`tf.compat.v1.ConfigProto`](https://www.tensorflow.org/api_docs/python/tf/compat/v1/ConfigProto) to configure Grappler graph optimization of the TensorFlow graphs. 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. |\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n| Returns ------- ||\n|---|---|\n| A `computation_base.Computation` that can generate state for a computation. ||\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n| Raises ------ ||\n|-------------|------------------------------------------|\n| `TypeError` | If the arguments are of the wrong types. |\n\n\u003cbr /\u003e"]]
View source on GitHub