Rename aesara.tensor.nnet.opt to aesara.tensor.nnet.rewriting

Author: brandonwillard

aesara/tensor/nnet/__init__.py

@@ -1,6 +1,6 @@
 import warnings
 
-import aesara.tensor.nnet.opt
+import aesara.tensor.nnet.rewriting
 from aesara.tensor.nnet.abstract_conv import (
     abstract_conv2d,
     conv2d,

aesara/tensor/nnet/opt.py renamed to aesara/tensor/nnet/rewriting.py

@@ -58,7 +58,7 @@ def local_inplace_sparse_block_gemv(fgraph, node):
     "fast_run",
     "inplace",
     position=60,
-)  # DEBUG
+)
 
 
 @node_rewriter([SparseBlockOuter], inplace=True)
@@ -82,7 +82,7 @@ def local_inplace_sparse_block_outer(fgraph, node):
     "fast_run",
     "inplace",
     position=60,
-)  # DEBUG
+)
 
 
 # Conv opts

aesara/tensor/nnet/sigm.py

@@ -1,6 +1,4 @@
 """
-Ops and optimizations: sigmoid, softplus.
-
 These functions implement special cases of exp and log to improve numerical
 stability.
 
@@ -98,7 +96,6 @@ def c_code_cache_version():
 pprint.assign(ultra_fast_sigmoid, printing.FunctionPrinter(["ultra_fast_sigmoid"]))
 
 
-# @opt.register_uncanonicalize
 @node_rewriter(None)
 def local_ultra_fast_sigmoid(fgraph, node):
     """
@@ -158,7 +155,6 @@ def hard_sigmoid(x):
     return x
 
 
-# @opt.register_uncanonicalize
 @node_rewriter([sigmoid])
 def local_hard_sigmoid(fgraph, node):
     if isinstance(node.op, Elemwise) and node.op.scalar_op == scalar_sigmoid: