File tree Expand file tree Collapse file tree 4 files changed +8
-8
lines changed Expand file tree Collapse file tree 4 files changed +8
-8
lines changed Original file line number Diff line number Diff line change 11import jax
22
33from pytensor .link .jax .dispatch import jax_funcify
4- from pytensor .tensor .signal .conv import Conv1d
4+ from pytensor .tensor .signal .conv import Convolve1d
55
66
7- @jax_funcify .register (Conv1d )
7+ @jax_funcify .register (Convolve1d )
88def jax_funcify_Conv1d (op , node , ** kwargs ):
99 mode = op .mode
1010
Original file line number Diff line number Diff line change 22
33from pytensor .link .numba .dispatch import numba_funcify
44from pytensor .link .numba .dispatch .basic import numba_njit
5- from pytensor .tensor .signal .conv import Conv1d
5+ from pytensor .tensor .signal .conv import Convolve1d
66
77
8- @numba_funcify .register (Conv1d )
8+ @numba_funcify .register (Convolve1d )
99def numba_funcify_Conv1d (op , node , ** kwargs ):
1010 mode = op .mode
1111
Original file line number Diff line number Diff line change 1515 from pytensor .tensor import TensorLike
1616
1717
18- class Conv1d (Op ):
18+ class Convolve1d (Op ):
1919 __props__ = ("mode" ,)
2020 gufunc_signature = "(n),(k)->(o)"
2121
@@ -129,4 +129,4 @@ def convolve1d(
129129 )
130130 mode = "valid"
131131
132- return cast (TensorVariable , Blockwise (Conv1d (mode = mode ))(in1 , in2 ))
132+ return cast (TensorVariable , Blockwise (Convolve1d (mode = mode ))(in1 , in2 ))
Original file line number Diff line number Diff line change 88from pytensor .graph import ancestors , rewrite_graph
99from pytensor .tensor import matrix , vector
1010from pytensor .tensor .blockwise import Blockwise
11- from pytensor .tensor .signal .conv import Conv1d , convolve1d
11+ from pytensor .tensor .signal .conv import Convolve1d , convolve1d
1212from tests import unittest_tools as utt
1313
1414
@@ -81,4 +81,4 @@ def test_convolve1d_batch_graph(mode):
8181 if var .owner is not None and isinstance (var .owner .op , Blockwise )
8282 ]
8383 # Check any Blockwise are just Conv1d
84- assert all (isinstance (node .op .core_op , Conv1d ) for node in blockwise_nodes )
84+ assert all (isinstance (node .op .core_op , Convolve1d ) for node in blockwise_nodes )
You can’t perform that action at this time.
0 commit comments