tf.lite.experimental.authoring.compatible
Оборачивает tf.function в вызываемую функцию с проверкой совместимости с TFLite.
tf.lite.experimental.authoring.compatible(
target=None, converter_target_spec=None, **kwargs
)
Используется в блокнотах
| Используется в руководстве |
| |
Пример:
@tf.lite.experimental.authoring.compatible
@tf.function(input_signature=[
tf.TensorSpec(shape=[None], dtype=tf.float32)
])
def f(x):
return tf.cosh(x)
result = f(tf.constant([0.0]))
# COMPATIBILITY WARNING: op 'tf.Cosh' require(s) "Select TF Ops" for model
# conversion for TensorFlow Lite.
# Op: tf.Cosh
# - tensorflow/python/framework/op_def_library.py:748
# - tensorflow/python/ops/gen_math_ops.py:2458
# - <stdin>:6
| Аргументы |
target | Функция tf.function для декорирования. |
converter_target_spec | target_spec параметра конвертера TFLite. |
**kwargs | Параметры ключевого слова класса декорирования _Compatible. |
| Возвращаемое значение |
Вызываемый объект типа tf.lite.experimental.authoring._Compatible. |