Refactoring Convolution; factor out convolutions from layers.py

ds-hwang · Dec 2, 2024 · 83fc2ce · 83fc2ce
1 parent c20387c
commit 83fc2ce
Show file tree

Hide file tree

Showing 28 changed files with 3,745 additions and 3,679 deletions.
diff --git a/axlearn/audio/subsamplers.py b/axlearn/audio/subsamplers.py
@@ -7,7 +7,8 @@
 
 from axlearn.common.base_layer import BaseLayer
 from axlearn.common.config import REQUIRED, Required, config_class
-from axlearn.common.layers import BaseNormalizationLayer, Conv2DWith1DPadding, get_activation_fn
+from axlearn.common.convolution import Conv2DWith1DPadding
+from axlearn.common.layers import BaseNormalizationLayer, get_activation_fn
 from axlearn.common.module import Module
 from axlearn.common.utils import Tensor
 

diff --git a/axlearn/common/conformer.py b/axlearn/common/conformer.py
@@ -30,9 +30,9 @@
 )
 from axlearn.common.base_layer import BaseLayer
 from axlearn.common.config import REQUIRED, InstantiableConfig, Required, config_class
+from axlearn.common.convolution import Conv1D
 from axlearn.common.layers import (
     BatchNorm,
-    Conv1D,
     Dropout,
     GroupNorm,
     LayerNorm,