Move sigmoid opt to math_opt

aesara/tensor/basic_opt.py

@@ -4069,7 +4069,7 @@ def local_flatten_lift(fgraph, node):
     Flatten(UnaryElemwise(x)) -> UnaryElemwise(Flatten(x))
 
     This optimization is needed by optimization
-    nnet/sigm.py:log1msigm_to_softplus to get applied when there is a flatten.
+    log1msigm_to_softplus to get applied when there is a flatten.
 
     """
     if (
@@ -4295,7 +4295,7 @@ def local_reshape_lift(fgraph, node):
     Reshape(UnaryElemwise(x)) -> UnaryElemwise(Reshape(x))
 
     This optimization is needed by optimization
-    nnet/sigm.py:log1msigm_to_softplus to get applied when there is a reshape.
+    log1msigm_to_softplus to get applied when there is a reshape.
 
     """
     if (

tests/tensor/nnet/test_sigm.py

@@ -9,16 +9,15 @@ from aesara.scalar import Softplus
 from aesara.tensor import sigmoid, softplus
 from aesara.tensor.inplace import neg_inplace, sigmoid_inplace
 from aesara.tensor.math import clip, exp, log, mul, neg
-from aesara.tensor.nnet.sigm import (
+from aesara.tensor.math_opt import (
     compute_mul,
-    hard_sigmoid,
     is_1pexp,
     parse_mul_tree,
     perform_sigm_times_exp,
     register_local_1msigmoid,
     simplify_mul,
-    ultra_fast_sigmoid,
 )
+from aesara.tensor.nnet.sigm import hard_sigmoid, ultra_fast_sigmoid
 from aesara.tensor.shape import Reshape
 from aesara.tensor.type import fmatrix, matrix, scalar, vector, vectors
 from tests import unittest_tools as utt