tff.program.MemoryReleaseManager

A tff.program.ReleaseManager that releases values to memory.

Inherits From: ReleaseManager

A tff.program.MemoryReleaseManager is a utility for releasing values from a federated program to memory and is used to release values from platform storage to customer storage in a federated program.

Values are released to memory as Python objects. When the value is released, if the value is a value reference or a structure containing value references, each value reference is materialized.

Methods

release

View source

Releases value from a federated program.

Args
value A tff.program.ReleasableStructure to release.
key A hashable value used to reference the released value.

remove_all

View source

Removes all program states.

values

View source

Returns an collections.OrderedDict of all keys and released values.