tff.framework.serialize_type

Serializes 'type_spec' as a pb.Type.

View aliases

Main aliases

tff.types.serialize_type

tff.framework.serialize_type(
    type_spec: Optional[tff.types.Type]
) -> Optional[pb.Type]

Note: Currently only serialization for tensor, named tuple, sequence, and function types is implemented.

Args
type_spec	A computation_types.Type, or None.

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

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