tff.framework.ConcreteComputation

View source on GitHub

A representation of a pb.Computation in the tff.Computation interface.

Inherits From: Computation, TypedObject

tff.framework.ConcreteComputation(
    computation_proto: pb.Computation,
    context_stack: tff.framework.ContextStack,
    annotated_type: Optional[tff.types.FunctionType] = None
)

This implementation exposes methods to retrieve the backing pb.Computation, as well as the Python representation of this protocol buffer represented by an instance of building_blocks.ComputationBuildingBlock. Leverages the implementation of __call__ inherited from function_utils.ConcreteFunction to pass self to the currently installed context.

Args
computation_proto	The protocol buffer that represents the computation, an instance of pb.Computation.
context_stack	The context stack to use.
annotated_type	Optional, type information with additional annotations that replaces the information in computation_proto.type.

Raises
TypeError	if annotated_type is not None and is not compatible with computation_proto.type.

Attributes
type_signature	Returns the TFF type of this object (an instance of tff.Type).

Methods

from_building_block

View source

@classmethod
from_building_block(
    building_block: tff.framework.ComputationBuildingBlock
) -> 'ConcreteComputation'

Converts a computation building block to a computation impl.

get_proto

View source

@classmethod
get_proto(
    value: 'ConcreteComputation'
) -> pb.Computation

to_building_block

View source

to_building_block()

to_compiled_building_block

View source

to_compiled_building_block()

with_type

View source

@classmethod
with_type(
    value: 'ConcreteComputation',
    type_spec: tff.types.Type
) -> 'ConcreteComputation'

__call__

View source

__call__(
    *args, **kwargs
)

Invokes the computation with the given arguments in the given context.

Args
*args	The positional arguments.
**kwargs	The keyword-based arguments.

Returns
The result of invoking the computation, the exact form of which depends on the context.

__eq__

View source

__eq__(
    other: Any
) -> bool

Return self==value.