tff.framework.SizingExecutorFactory

Stay organized with collections Save and categorize content based on your preferences.

A executor factory holding an executor per cardinality.

Inherits From: ResourceManagingExecutorFactory, ExecutorFactory

tff.framework.SizingExecutorFactory(
    executor_stack_fn: Callable[[executor_factory.CardinalitiesType], tuple[executor_base.Executor,
        list[sizing_executor.SizingExecutor]]]
)

executor_stack_fn Similar to base class but the second return value of the callable is used to expose the SizingExecutors.

Methods

clean_up_executor

View source

clean_up_executor(
    cardinalities: tff.framework.CardinalitiesType
)

Calls close on constructed executors, resetting internal cache.

If a caller holds a name bound to any of the executors returned from create_executor with the specified cardinalities, 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.

Args
cardinalities The cardinalities of the executor to be cleaned up.

create_executor

View source

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

See base class.

get_size_info

View source

get_size_info() -> tff.framework.SizeInfo

Returns information about the transferred data of each SizingExecutor.

Returns the history of broadcast and aggregation for each executor as well as the number of aggregated bits that has been passed through.

Returns
An instance of SizeInfo.