Module: tfa

View source on GitHub

Useful extra functionality for TensorFlow maintained by SIG-addons.

Modules

activations module: A module containing activation routines.

callbacks module: A module containing callbacks that conform to Keras API.

image module: Image manipulation ops.

layers module: Additional layers that conform to Keras API.

losses module: Additional losses that conform to Keras API.

metrics module: A module containing metrics that conform to Keras API.

optimizers module: Additional optimizers that conform to Keras API.

rnn module: Customized RNN cells.

seq2seq module: Ops for building neural network sequence to sequence decoders and losses.

text module: Text-processing ops.