View source on GitHub |
Serializes 'tff.Computation' as a pb.Computation.
tff.framework.serialize_computation(
computation: tff.Computation
) -> pb.Computation
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. |