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 tff.program.ReleasableStructure to release.
|
type_signature
|
The tff.Type of value.
|
key
|
A hashable value used to reference the released value.
|
values
values() -> collections.OrderedDict[Hashable, tuple[release_manager.ReleasableStructure,
computation_types.Type]]
Returns an collections.OrderedDict of all keys and released values and types.