tff.framework.serialize_type

View source on GitHub

Serializes 'type_spec' as a pb.Type.

type_spec Either an instance of computation_types.Type, or something convertible to it by computation_types.to_type(), or None.

The corresponding instance of pb.Type, or None if the argument was None.

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