tff.framework.ResourceManagingExecutorFactory

Implementation of executor factory holding an executor per cardinality.

Inherits From: ExecutorFactory

tff.framework.ResourceManagingExecutorFactory(
    executor_stack_fn: Callable[[executor_factory.CardinalitiesType], executor_base.Executor],
    ensure_closed: Optional[Sequence[executor_base.Executor]] = None
)

Args
executor_stack_fn	Callable taking a mapping from placement_literals.PlacementLiteral to integers, and returning an executor_base.Executor. The returned executor will be configured to handle these cardinalities.
ensure_closed	Optional sequence of executor_base.Excutors which should always be closed on a clean_up_executors call. Defaults to empty.

Methods

clean_up_executors

View source

clean_up_executors()

Calls close on all constructed executors, resetting internal cache.

If a caller holds a name bound to any of the executors returned from create_executor, this executor should be assumed to be in an invalid state, and should not be used after this method is called. Instead, callers should again invoke create_executor.

create_executor

View source

create_executor(
    cardinalities: executor_factory.CardinalitiesType
) -> tff.framework.Executor

Constructs or gets existing executor.

Returns a previously-constructed executor if this method has already been invoked with cardinalities. If not, invokes self._executor_stack_fn with cardinalities and returns the result.

Args
cardinalities	dict with placement_literals.PlacementLiteral keys and integer values, specifying the population size at each placement. The executor stacks returned from this method are not themselves polymorphic; a concrete stack must have fixed sizes at each placement.

Returns
Instance of executor_base.Executor as described above.