tff.framework.FederatingExecutor

An executor for handling federated types and intrinsics.

Inherits From: Executor

tff.framework.FederatingExecutor(
    strategy_factory: Callable[['FederatingExecutor'], tff.framework.FederatingStrategy],
    unplaced_executor: tff.framework.Executor
)

At a high-level, this executor is responsible for handling federated types and federated intrinsics and delegating work to an underlying collection of target executors associated with individual placements.

The responsibility for handling federated types and federated intrinsics is delegated to a FederatingStrategy, specifically:

  • which federated intrinsics are implemented
  • how federated types and federated intrinsics are implemented
  • how federated values are represented in the execution stack
  • how work is delegated to the target executors
  • which placements are implemented

The reponsibility for handling unplaced types, computations, and processing is delegated to an unplaced_executor.

Expressed in a different way, this executor integrates a FederatedStrategy with an execution stack, providing a way to inject the logic for handling federated types and intrinsics into the execution stack.

strategy_factory A function that accepts an instance of a FederatingExecutor and returns a newly constructed FederatingStrategy to use to handle federated types and intrinsics.
unplaced_executor An executor_base.Executor to use to handle unplaced types, computations, and processing.

Methods

close

View source

close()

Release resources associated with this Executor, if any.

If the executor has one or more target Executors, implementation of this method must close them.

create_call

View source

create_call(
    *fn_args, **fn_kwargs
)

create_selection

View source

create_selection(
    *fn_args, **fn_kwargs
)

create_struct

View source

create_struct(
    *fn_args, **fn_kwargs
)

create_value

View source

create_value(
    *fn_args, **fn_kwargs
)