Skip to content

Commit d239486

Browse files
committed
Implements nin and nout for element-wise funcs
`nin` and `nout` properties return the number of arguments to the function treated as inputs or outputs, respectively
1 parent 5ec9fd5 commit d239486

File tree

1 file changed

+28
-0
lines changed

1 file changed

+28
-0
lines changed

dpctl/tensor/_elementwise_common.py

+28
Original file line numberDiff line numberDiff line change
@@ -93,6 +93,20 @@ def get_type_result_resolver_function(self):
9393
"""
9494
return self.result_type_resolver_fn_
9595

96+
@property
97+
def nin(self):
98+
"""
99+
Returns the number of arguments treated as inputs.
100+
"""
101+
return 1
102+
103+
@property
104+
def nout(self):
105+
"""
106+
Returns the number of arguments treated as outputs.
107+
"""
108+
return 1
109+
96110
@property
97111
def types(self):
98112
"""Returns information about types supported by
@@ -531,6 +545,20 @@ def get_type_promotion_path_acceptance_function(self):
531545
"""
532546
return self.acceptance_fn_
533547

548+
@property
549+
def nin(self):
550+
"""
551+
Returns the number of arguments treated as inputs.
552+
"""
553+
return 2
554+
555+
@property
556+
def nout(self):
557+
"""
558+
Returns the number of arguments treated as outputs.
559+
"""
560+
return 1
561+
534562
@property
535563
def types(self):
536564
"""Returns information about types supported by

0 commit comments

Comments
 (0)