Skip to content

Commit 46e3e72

Browse files
author
Ian Schweer
committed
Better name
1 parent 4ebdd15 commit 46e3e72

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pytensor/link/pytorch/dispatch/elemwise.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ def elemwise_fn(*inputs):
3535
return base_fn(*inputs)
3636

3737
elif isinstance(scalar_op, ScalarLoop):
38-
return elemwise_scalar_loop(base_fn, op, node, **kwargs)
38+
return elemwise_ravel_fn(base_fn, op, node, **kwargs)
3939

4040
else:
4141

@@ -182,7 +182,7 @@ def softmax_grad(dy, sm):
182182
return softmax_grad
183183

184184

185-
def elemwise_scalar_loop(base_fn, op, node, **kwargs):
185+
def elemwise_ravel_fn(base_fn, op, node, **kwargs):
186186
"""
187187
ScalarLoop + Elemwise is too common
188188
to not work, but https://github.com/pymc-devs/pytensor/issues/1031,

0 commit comments

Comments
 (0)