Context
context_base.Context는 AST를 구성, 컴파일 또는 실행할 수 있는 환경입니다.
이 API는 Executor가 실행에 사용되지 않을 때 사용해야 하는 하위 수준의 추상화를 정의합니다. Reference 백엔드는 이 수준에서 통합됩니다.
ExecutionContext
execution_context.ExecutionContext는 컴파일 함수를 사용하여 계산을 컴파일하고 Executor를 사용하여 계산을 실행하는 context_base.Context입니다.
이 API는 Executor가 실행에 사용될 때 사용해야 하는 상위 수준의 추상화를 정의합니다. native 및 IREE 백엔드는 이 수준에서 통합됩니다.
FederatedComputationContext
federated_computation_context.FederatedComputationContext는 페더레이션 계산을 구성하는 context_base.Context입니다. 이 컨텍스트는 computations.federated_computation 데코레이터로 데코레이팅된 Python 함수를 추적하는 데 사용됩니다.
TensorFlowComputationContext
tensorflow_computation_context.TensorFlowComputationContext는 TensorFlow 계산을 구성하는 context_base.Context입니다. 이 컨텍스트는computations.tf_computation 데코레이터로 데코레이팅된 Python 함수를 직렬화하는 데 사용됩니다.
ContextStack
context_stack_base.ContextStack은 Context 스택과 상호 작용하기 위한 데이터 구조입니다.
TFF가 AST를 구성, 컴파일 또는 실행하는 데 사용할 컨텍스트를 다음과 같이 설정할 수 있습니다.
set_default_context.set_default_context를 호출하여 기본 컨텍스트를 설정합니다. 이 API는 종종 계산을 컴파일하거나 실행할 컨텍스트를 설치하는 데 사용됩니다.
get_context_stack.get_context_stack을 호출하여 현재 컨텍스트 스택을 가져온 다음 context_stack_base.ContextStack.install을 호출하여 스택의 맨 위에 컨텍스트를 임시로 설치합니다. 예를 들어, calculations.federated_computation 및 calculations.tf_computation 데코레이터는 데코레이팅된 함수가 추적되는 동안 해당 컨텍스트를 현재 컨텍스트 스택으로 푸시합니다.
ContextStackImpl
context_stack_impl.ContextStackImpl은 공통 스레드 로컬 스택으로 구현되는 context_stack_base.ContextStack입니다.