We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 4ebdd15 commit 46e3e72Copy full SHA for 46e3e72
pytensor/link/pytorch/dispatch/elemwise.py
@@ -35,7 +35,7 @@ def elemwise_fn(*inputs):
35
return base_fn(*inputs)
36
37
elif isinstance(scalar_op, ScalarLoop):
38
- return elemwise_scalar_loop(base_fn, op, node, **kwargs)
+ return elemwise_ravel_fn(base_fn, op, node, **kwargs)
39
40
else:
41
@@ -182,7 +182,7 @@ def softmax_grad(dy, sm):
182
return softmax_grad
183
184
185
-def elemwise_scalar_loop(base_fn, op, node, **kwargs):
+def elemwise_ravel_fn(base_fn, op, node, **kwargs):
186
"""
187
ScalarLoop + Elemwise is too common
188
to not work, but https://github.com/pymc-devs/pytensor/issues/1031,
0 commit comments