Merge pull request #3712 from carlosgmartin:conv_lstm_cell_docs

PiperOrigin-RevId: 615514457
google · Mar 13, 2024 · 3da44bc · 3da44bc
2 parents 50b589f + 1c9cdc1
commit 3da44bc
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/docs/api_reference/flax.linen/layers.rst b/docs/api_reference/flax.linen/layers.rst
@@ -122,6 +122,10 @@ Recurrent
   :module: flax.linen
   :class: OptimizedLSTMCell
 
+.. flax_module::
+  :module: flax.linen
+  :class: ConvLSTMCell
+
 .. flax_module::
   :module: flax.linen
   :class: SimpleCell