tf.types.experimental.SupportsTracingProtocol

Протокол, позволяющий пользовательским классам управлять повторной трассировкой tf.function.

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

Методы

__tf_tracing_type__

Посмотреть исходный код

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

Возвращает тип трассировки этого объекта.

Тип трассировки используется для формирования сигнатуры tf.function при трассировке и для сопоставления аргументов с существующими сигнатурами. При вызове объекта Function функция tf.function проверяет тип трассировки аргументов вызова. Если существует сигнатура с совпадающим типом, она будет использована. В противном случае трассируется новая функция, и в её сигнатуре будет использоваться тип трассировки аргументов вызова.