Skip to content

Commit 4ae5971

Browse files
committed
Export useful helpers from elemwise.
1 parent 3ef3040 commit 4ae5971

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

pygpu/elemwise.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,8 @@
44
from . import gpuarray
55
from ._elemwise import GpuElemwise, arg
66

7-
__all__ = ['GpuElemwise', 'elemwise1', 'elemwise2', 'ielemwise2', 'compare']
7+
__all__ = ['GpuElemwise', 'arg', 'as_argument',
8+
'elemwise1', 'elemwise2', 'ielemwise2', 'compare']
89

910

1011
def _dtype(o):

0 commit comments

Comments
 (0)