Module: tfa

Useful extra functionality for TensorFlow maintained by SIG-addons.

Modules

activations module: Additional activation functions.

callbacks module: Additional callbacks that conform to Keras API.

image module: Additional image manipulation ops.

layers module: Additional layers that conform to Keras API.

losses module: Additional losses that conform to Keras API.

metrics module: Additional metrics that conform to Keras API.

optimizers module: Additional optimizers that conform to Keras API.

options module

rnn module: Additional RNN cells that corform to Keras API.

seq2seq module: Additional layers for sequence to sequence models.

text module: Additional text-processing ops.

types module: Types for typing functions signatures.

Functions

register_all(...): Register TensorFlow Addons' objects in TensorFlow global dictionaries.