Supported TensorFlow Keras Layers
The following Keras layers are currently supported for model compression and optimization:
Core Layers
- Densetf.keras.layers.Dense
- Conv2Dtf.keras.layers.Conv2D
- InputLayertf.keras.layers.InputLayer
- Flattentf.keras.layers.Flatten
- Reshapetf.keras.layers.Reshape
Recurrent Layers
- RNNtf.keras.layers.RNN
- LSTMtf.keras.layers.LSTM
- GRUtf.keras.layers.GRU
Pooling Layers
- AveragePooling1Dtf.keras.layers.AveragePooling1D
- AveragePooling2Dtf.keras.layers.AveragePooling2D
- AveragePooling3Dtf.keras.layers.AveragePooling3D
- MaxPool2Dtf.keras.layers.MaxPool2D
- MinPool2Dtf.keras.layers.MinPool2D
- Averagetf.keras.layers.Average
Activation Functions
- ReLUtf.keras.activations.relu
- Leaky ReLUtf.keras.activations.leaky_relu
- Tanhtf.keras.activations.tanh
- Sigmoidtf.keras.activations.sigmoid
- Swishtf.keras.activations.swish
- ELUtf.keras.activations.elu
- Softmaxtf.keras.activations.softmax
Other Layers
- Dropouttf.keras.layers.Dropout
- Activationtf.keras.layers.Activation
- ActivityRegularizationtf.keras.layers.ActivityRegularization
- Addtf.keras.layers.Add
Coming Soon
Support for additional layer types is in development. Please check back for updates or contact our team for specific layer support requests.