PiperOrigin-RevId: 338088240

PiperOrigin-RevId: 338087663

PiperOrigin-RevId: 338076995

PiperOrigin-RevId: 337964148

```
TypeError: Expected tensorflow.python.ops.variables.Variable, found tensorflow.python.keras.engine.base_layer_utils.TrackableWeightHandler.
```
PiperOrigin-RevId: 337961843

PiperOrigin-RevId: 337958997

Instead insert a `tf.function` decorator before (underneath) the
`tff.tf_computation` decorator, and enter a `tf.init_scope` context during
`tf.Variable` creation. The `tf.init_scope` hoists variable creation out of the
`tf.function` to the nearest graph scope (provided by `tff.tf_comptuation`),
which would otherwise result in an error from the the `tf.function` wrapper.

More details on `tf.Variable` in `tf.function`:
https://www.tensorflow.org/guide/function#variables

Longer term, it may be nice to fold this automatic wrapping into the
`tff.tf_computation` decorator, or re-introduce as `tff.tf2_computation`,
however this requires all users to know they must use `tf.init_scope` during
variable creation. Inside TFF packages (e.g.  `tff.learning`) we can handle
this for them, but its an extra point of concern for custom algorithm writers.

PiperOrigin-RevId: 337935371

Using `attr.s` classes as TFF type specifications has usability issues when used in conjunction with python static type checking. Mixing python type checking with TF and TFF types does not work well. This functionality will be removed in a future release.

PiperOrigin-RevId: 337578239

This enables support for Python 3.8, which changed behavior of `_asdict` to
return regular `dict` types.
https://docs.python.org/3/library/collections.html#collections.somenamedtuple._asdict

PiperOrigin-RevId: 337559550

This change moves the `apply` method inside the `EstimationProcess` class, and renames it as `map`. It also updates the tests to be consistent with the pattern in the other processes in templates.

PiperOrigin-RevId: 337424574

Adds DifferentiallyPrivateFactory, an AggregationProcessFactory for DPQueries from tensorflow_privacy.

PiperOrigin-RevId: 337400971

This is a sibling
to CanonicalForm which is intended to represent a
new task type which involves only server work +
broadcast + client work with no final aggregation.

My hope is to reuse much of the existing CanonicalForm
compilation infrastructure for compiling to this form
(that is, sharing much of canonical_form_utils.py
between both forms). An alternative approach would be
to slightly extend and re-use CanonicalForm by adding
a client output and checking to ensure that the arguments
to the aggregation functions are empty structures. I'm
not strongly opposed to that formulation, but have
developed a weak preference for this approach in order
to reduce overloading and layering of complexity on top
of the existing system. My hope is that the approach taken
in this CL will encourage work towards easier
development of varied computation "shape" definitions.

PiperOrigin-RevId: 337184931

PiperOrigin-RevId: 337038042

Change the shape of "class" from [] to [1]

PiperOrigin-RevId: 336980750

Much of the existing complication was unnecessary
since the change to make `intrinsics.federated_zip`
recursive.

PiperOrigin-RevId: 336967098

Fixes pytype mismatch in optimizer_utils, adding extra typing info in federated_averaging and federated_sgd.

PiperOrigin-RevId: 336957857

PiperOrigin-RevId: 336901411

Also fix typo.

PiperOrigin-RevId: 336891259

PiperOrigin-RevId: 336798271

Update pydoc to use `attributes` section.

PiperOrigin-RevId: 336735972

PiperOrigin-RevId: 336681767

PiperOrigin-RevId: 336557256

PiperOrigin-RevId: 336383978

PiperOrigin-RevId: 336355194

Add `CanonicalForm.securely_aggregates_tensors` API to check whether a given `CanonicalForm` uses secure aggregation.

PiperOrigin-RevId: 336332257

Implements a minimal ExecutorService::Execute mock that can be configured with a return value or raise.

PiperOrigin-RevId: 336329041

PiperOrigin-RevId: 336322146

PiperOrigin-RevId: 336313295

get_estimate returns the result of some computation performed on the process state.

PiperOrigin-RevId: 336199570

PiperOrigin-RevId: 336189982

PiperOrigin-RevId: 336189398

PiperOrigin-RevId: 336178759

PiperOrigin-RevId: 336163849

PiperOrigin-RevId: 336129461

PiperOrigin-RevId: 336128007

PiperOrigin-RevId: 335965990

This allows TestCase to become our "uber testcase"
and pick up methods that depend on other behavior
in api/, such as type equivalence checks.

PiperOrigin-RevId: 335919774

PiperOrigin-RevId: 335907972

PiperOrigin-RevId: 335905651

Intention here is to investigate how much overhead is introduced by the synchronous RPC client.

PiperOrigin-RevId: 335882989