tff.framework.serialize_computation

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

computation An instance of tff.Computation.

The corresponding instance of pb.Computation.

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