View source on GitHub
|
A tff.program.ReleaseManager that releases values to memory.
Inherits From: ReleaseManager
tff.program.MemoryReleaseManager()
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
release(
value, type_signature, key
)
Releases value from a federated program.
| Args | |
|---|---|
value
|
A materialized value, a value reference, or a structure of materialized values and value references representing the value to release. |
type_signature
|
The tff.Type of value.
|
key
|
A hashable value used to reference the released value.
|
values
values() -> OrderedDict[Hashable, Tuple[Any, computation_types.Type]]
Returns an OrderedDict of all keys and released values and types.