Lines Matching defs:args
87 def my_fallback(op, *args, **kwargs):
137 def first_fallback(keyset, op, *args, **kwargs):
151 def second_fallback(op, *args, **kwargs):
186 def my_fallback(op, *args, **kwargs):
209 def my_neg(*args, **kwargs):
228 def my_mul(*args, **kwargs):
301 def my_sum(*args, **kwargs):
325 def inverted_where(*args, **kwargs):
357 def fast_gelu(*args, **kwargs):
387 def clipped_exp(*args, **kwargs):
418 def buggy_add(*args, **kwargs):
444 def my_sum(*args, **kwargs):
464 def my_sum(*args, **kwargs):
474 def my_sum_zt(*args, **kwargs):
489 def my_sum(*args, **kwargs):
502 def my_sum4(*args, **kwargs):
513 def my_sum3(*args, **kwargs):
529 def _op(*args, **kwargs):
595 def _check_is_functional_variant(self, mutable_op, functional_op, args): argument
628 def fn(*args):
915 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): argument
935 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): argument
1004 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): argument
1013 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): argument
1109 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): argument
1117 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): argument
1176 def __new__(cls, elem, *args, **kwargs):
1191 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): argument
1232 def __new__(cls, elem, *args, **kwargs):
1247 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): argument
1295 def __new__(cls, elem, *args, **kwargs):
1318 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): argument
1352 def __new__(cls, elem, *args, **kwargs):
1365 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): argument
1458 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): argument
1468 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): argument
1473 def __torch_dispatch__(self, func, types, args=(), kwargs=None): argument
1477 def __torch_dispatch__(self, func, types, args=(), kwargs=None): argument
1506 def __torch_dispatch__(self, func, types, args=(), kwargs=None): argument
1530 def __torch_dispatch__(self, func, types, args=(), kwargs=None): argument
1552 def __torch_dispatch__(self, func, types, args=(), kwargs=None): argument
1569 def __torch_dispatch__(self, func, types, args=(), kwargs=None): argument
1586 def __torch_dispatch__(self, func, types, args=(), kwargs=None): argument
1603 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): argument
1607 def __torch_dispatch__(self, func, types, args=(), kwargs=None): argument
1623 def __torch_dispatch__(self, func, types, args=(), kwargs=None): argument
1653 def __torch_dispatch__(self, func, types, args=(), kwargs=None): argument
1667 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): argument
1709 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): argument
1721 def __torch_dispatch__(self, func, types, args=(), kwargs=None): argument
1735 def __torch_dispatch__(self, func, types, args=(), kwargs=None): argument
1801 def __torch_dispatch__(self, func, types, args=(), kwargs=None): argument
1824 def __torch_dispatch__(self, func, types, args=(), kwargs=None): argument
1848 def __new__(cls, elem, *args, **kwargs):
1858 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): argument
1889 def __new__(cls, elem, *args, **kwargs):
1902 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): argument
1967 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): argument
1985 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): argument
2002 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): argument
2022 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): argument
2057 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): argument
2111 def __torch_dispatch__(cls, func, types, args, kwargs): argument
2122 def __torch_dispatch__(cls, func, types, args, kwargs): argument
2135 def __torch_dispatch__(cls, func, types, args, kwargs): argument
2168 def __torch_dispatch__(cls, func, types, args, kwargs): argument
2211 def __torch_dispatch__(cls, func, types, args, kwargs): argument
2222 def __torch_dispatch__(cls, func, types, args, kwargs): argument
2235 def __torch_dispatch__(cls, func, types, args, kwargs): argument
2267 def __torch_dispatch__(cls, func, types, args, kwargs): argument
2278 def __torch_dispatch__(cls, func, types, args, kwargs): argument
2294 def __torch_function__(cls, *args, **kwargs):
2308 def __new__(cls, *args, **kwargs):
2315 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): argument
2355 def __torch_dispatch__(cls, func, types, args, kwargs): argument
2366 def __torch_dispatch__(cls, func, types, args, kwargs): argument
2379 def __torch_dispatch__(cls, func, types, args, kwargs): argument
2407 def __torch_dispatch__(cls, func, types, args, kwargs): argument
2420 def __torch_dispatch__(cls, func, types, args, kwargs): argument
2435 def __torch_dispatch__(cls, func, types, args, kwargs): argument
2481 def __torch_dispatch__(cls, func, types, args, kwargs): argument
2515 def __torch_dispatch__(cls, func, types, args, kwargs): argument
2541 def __torch_dispatch__(cls, func, types, args, kwargs): argument
2552 def __torch_dispatch__(cls, func, types, args, kwargs): argument
2565 def __torch_dispatch__(cls, func, types, args, kwargs): argument
2598 def _test_wrapper_subclass_aliasing(self, op, args, kwargs): argument