tf.types.experimental.SupportsTracingProtocol

A protocol allowing custom classes to control tf.function retracing.

tf.types.experimental.SupportsTracingProtocol(
    *args, **kwargs
)

Methods

__tf_tracing_type__

View source

@abc.abstractmethod
__tf_tracing_type__(
    context: TracingContext
) -> tf.types.experimental.TraceType

Returns the tracing type of this object.

The tracing type is used to build the signature of a tf.function when traced, and to match arguments with existing signatures. When a Function object is called, tf.function looks at the tracing type of the call arguments. If an existing signature of matching type exists, it will be used. Otherwise, a new function is traced, and its signature will use the tracing type of the call arguments.

Args
context a context reserved for internal/future usage.

Returns
The tracing type of this object.