tf.lite.experimental.authoring.compatible

Wraps tf.function into a callable function with TFLite compatibility checking.

tf.lite.experimental.authoring.compatible(
    target=None, converter_target_spec=None, **kwargs
)

Used in the notebooks

Used in the guide

Example:

@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 A tf.function to decorate.
converter_target_spec target_spec of TFLite converter parameter.
**kwargs The keyword arguments of the decorator class _Compatible.

A callable object of tf.lite.experimental.authoring._Compatible.