Introducing Google AI Edge Portal : Benchmark Edge AI at scale. Sign-up to request access during private preview.

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

TFLite Authoring Tool

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

Args

target

A tf.function to decorate.

converter_target_spec

target_spec of TFLite converter parameter.

**kwargs

The keyword arguments of the decorator class _Compatible.

Returns

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