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, key
)
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
remove_all() -> None
Removes all program states.
values
values() -> collections.OrderedDict[Hashable, release_manager.ReleasableStructure]
Returns an collections.OrderedDict of all keys and released values.