tff.framework.serialize_computation

Serializes 'tff.Computation' as a pb.Computation.

tff.framework.serialize_computation(
    computation: tff.Computation
) -> pb.Computation

Used in the notebooks

Used in the tutorials
Experimental support for JAX in TFF

Note: Currently only serialization for computation_impl.ConcreteComputation is implemented.

Args
computation	An instance of tff.Computation.

Returns
The corresponding instance of pb.Computation.

Raises
TypeError	If the argument is of the wrong type.
NotImplementedError	for computation variants for which serialization is not implemented.