add layers api to init

keras-team · Jan 27, 2024 · 3f26791 · 3f26791
1 parent 2636d9b
commit 3f26791
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/keras_cv/layers/__init__.py b/keras_cv/layers/__init__.py
@@ -141,3 +141,5 @@
 from keras_cv.layers.vit_det_layers import WindowedTransformerEncoder
 from keras_cv.layers.vit_det_layers import WindowPartitioning
 from keras_cv.layers.vit_layers import PatchingAndEmbedding
+from keras_cv.layers.video_swin_transformer_layers import WindowAttention3D
+from keras_cv.layers.video_swin_transformer_layers import SwinTransformerBlock3D