上下文

Context

context_base.Context 是一种可以构造、编译或执行 AST 的环境。

此 API 可定义在不使用执行器执行时应使用的低级抽象；参考后端在此级别集成。

ExecutionContext

execution_context.ExecutionContext 是使用编译函数来编译计算并使用执行器来执行计算的 context_base.Context。

此 API 定义了在使用执行器执行时应使用的高级抽象；原生后端在此级别集成。

FederatedComputationContext

federated_computation_context.FederatedComputationContext 是用于构造联合计算的 context_base.Context。此上下文用于跟踪使用 federated_computation.federated_computation 装饰器装饰的 Python 函数。

TensorFlowComputationContext

tensorflow_computation_context.TensorFlowComputationContext 是用于构造 TensorFlow 计算的 context_base.Context。此上下文用于对使用 tensorflow_computation.tf_computation 装饰器装饰的 Python 函数执行序列化。

ContextStack

context_stack_base.ContextStack 是用于与上下文堆栈进行交互的数据结构。

您可以通过以下方式设置 TFF 将用于构造、编译或执行 AST 的上下文：

• 调用 set_default_context.set_default_context 以设置默认上下文。此 API 常用于安装将编译或执行计算的上下文。

• 调用 get_context_stack.get_context_stack 以获取当前的上下文堆栈，然后调用 context_stack_base.ContextStack.install 以将上下文临时安装到堆栈顶部。例如，在跟踪装饰的函数的同时，federated_computation.federated_computation 和 tensorflow_computation.tf_computation 装饰器可将相应的上下文推送到当前上下文堆栈。

ContextStackImpl

context_stack_impl.ContextStackImpl 是作为常用线程局部堆栈实现的 context_stack_base.ContextStack。