diff options
Diffstat (limited to 'candle-pyo3/py_src/candle/functional/__init__.py')
-rw-r--r-- | candle-pyo3/py_src/candle/functional/__init__.py | 2 |
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 |