summaryrefslogtreecommitdiff
path: root/candle-pyo3/py_src/candle/functional/__init__.py
diff options
context:
space:
mode:
Diffstat (limited to 'candle-pyo3/py_src/candle/functional/__init__.py')
-rw-r--r--candle-pyo3/py_src/candle/functional/__init__.py2
1 files changed, 2 insertions, 0 deletions
diff --git a/candle-pyo3/py_src/candle/functional/__init__.py b/candle-pyo3/py_src/candle/functional/__init__.py
index efb246f0..095e54e4 100644
--- a/candle-pyo3/py_src/candle/functional/__init__.py
+++ b/candle-pyo3/py_src/candle/functional/__init__.py
@@ -1,7 +1,9 @@
# Generated content DO NOT EDIT
from .. import functional
+avg_pool2d = functional.avg_pool2d
gelu = functional.gelu
+max_pool2d = functional.max_pool2d
relu = functional.relu
silu = functional.silu
softmax = functional.softmax