| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _tensor.py | 23 handle_torch_function,
38 return handle_torch_function(wrapped, args, *args, **kwargs)
83 return handle_torch_function(Tensor.__deepcopy__, (self,), self, memo)
228 return handle_torch_function(Tensor.__reduce_ex__, (self,), self, proto)
245 return handle_torch_function(Tensor.storage, (self,), self)
500 return handle_torch_function(Tensor.__setstate__, (self,), self, state)
519 return handle_torch_function(
572 return handle_torch_function(
619 return handle_torch_function(Tensor.register_hook, (self,), self, hook)
674 return handle_torch_function(
[all …]
|
| H A D | functional.py | 13 handle_torch_function,
75 return handle_torch_function(broadcast_tensors, tensors, *tensors)
200 return handle_torch_function(
390 return handle_torch_function(einsum, operands, equation, *operands)
524 return handle_torch_function(meshgrid, tensors, *tensors, indexing=indexing)
682 return handle_torch_function(
949 return handle_torch_function(
1033 return handle_torch_function(
1315 return handle_torch_function(tensordot, (a, b), a, b, dims=dims, out=out)
1391 return handle_torch_function(cartesian_prod, tensors, *tensors)
[all …]
|
| H A D | _lowrank.py | 9 from torch.overrides import handle_torch_function, has_torch_function
144 return handle_torch_function(
252 return handle_torch_function(
|
| H A D | __init__.py | 740 return overrides.handle_torch_function(sym_not, (a,), a)
755 return overrides.handle_torch_function(sym_float, (a,), a)
770 return overrides.handle_torch_function(sym_int, (a,), a)
786 return overrides.handle_torch_function(sym_max, (a, b), a, b)
826 return overrides.handle_torch_function(sym_min, (a, b), a, b)
846 return overrides.handle_torch_function(fn, (a,), a)
881 return overrides.handle_torch_function(sym_ite, (b, t, f), b, t, f)
2037 return overrides.handle_torch_function(
|
| H A D | _lobpcg.py | 10 from torch.overrides import handle_torch_function, has_torch_function
510 return handle_torch_function(
|
| H A D | overrides.py | 246 torch.nn.functional.handle_torch_function,
268 handle_torch_function,
1579 return handle_torch_function(wrapped, relevant_args, *args, **kwargs)
1666 def handle_torch_function( function
|
| H A D | _ops.py | 424 return torch.overrides.handle_torch_function(
|
| /aosp_15_r20/external/pytorch/tools/autograd/templates/ |
| H A D | python_variable_methods.cpp | 70 return handle_torch_function(self, "_is_view", args); in THPVariable__is_view()
88 return handle_torch_function(self, "apply_", args.ptr()); in THPVariable_apply_()
112 return handle_torch_function(r, self, args, kwargs, THPVariableClass, "torch.Tensor"); in THPVariable_size()
146 return handle_torch_function(r, self, args, kwargs, THPVariableClass, "torch.Tensor"); in THPVariable_stride()
178 return handle_torch_function(self_, "get_device", args, nullptr); in THPVariable_get_device()
189 return handle_torch_function(self_, "has_names", args); in THPVariable_has_names()
201 return handle_torch_function(self_, "data_ptr", args); in THPVariable_data_ptr()
213 return handle_torch_function(self_, "storage_offset"); in THPVariable_storage_offset()
225 return handle_torch_function(self, "dim", args); in THPVariable_dim()
237 return handle_torch_function(self, "numel", args); in THPVariable_numel()
[all …]
|
| H A D | python_nn_functions.cpp | 47 …return handle_torch_function(r, args, kwargs, THPNNVariableFunctionsModule, "torch.nn", "_parse_to… in THPVariable__parse_to()
|
| /aosp_15_r20/external/pytorch/torch/nn/ |
| H A D | functional.py | 22 handle_torch_function,
477 return handle_torch_function(
522 return handle_torch_function(
596 return handle_torch_function(
638 return handle_torch_function(
699 return handle_torch_function(
727 return handle_torch_function(
789 return handle_torch_function(
817 return handle_torch_function(
879 return handle_torch_function(
[all …]
|
| H A D | init.py | 163 return torch.overrides.handle_torch_function(
190 return torch.overrides.handle_torch_function(
239 return torch.overrides.handle_torch_function(
500 return torch.overrides.handle_torch_function(
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | python_torch_functions_manual.cpp | 130 return handle_torch_function( in THPVariable_as_tensor()
181 return handle_torch_function( \
232 return handle_torch_function( in THPVariable_sparse_coo_tensor()
258 return handle_torch_function( in THPVariable_tensor()
283 return handle_torch_function( in THPVariable_get_device()
341 return handle_torch_function( in THPVariable_asarray()
439 return handle_torch_function( in THPVariable_nonzero()
477 return handle_torch_function( in THPVariable_numel()
|
| H A D | init.cpp | 877 return handle_torch_function( in set_grad_enabled()
931 return handle_torch_function( in set_multithreading_enabled()
969 return handle_torch_function( in set_view_replay_enabled()
|
| H A D | python_variable_indexing.cpp | 41 handle_torch_function(self, "__len__")); in THPVariable_length()
|
| /aosp_15_r20/external/pytorch/torch/utils/ |
| H A D | backend_registration.py | 4 handle_torch_function,
122 …return handle_torch_function(wrap_tensor_backend.__get__, (self,), self) # type: ignore[attr-defi…
146 …return handle_torch_function(wrap_tensor_to, (self,), self, device=device, non_blocking=False, **k…
|
| /aosp_15_r20/external/pytorch/torch/csrc/utils/ |
| H A D | python_arg_parser.cpp | 163 return handle_torch_function( in handle_torch_function_getter()
181 handle_torch_function( in handle_torch_function_setter()
189 handle_torch_function( in handle_torch_function_setter()
226 auto handle_torch_function( in handle_torch_function() function
593 auto handle_torch_function( in handle_torch_function() function
617 auto handle_torch_function( in handle_torch_function() function
624 return handle_torch_function( in handle_torch_function()
|
| H A D | python_arg_parser.h | 1181 auto handle_torch_function(
1191 auto handle_torch_function(
1200 auto handle_torch_function(
|
| /aosp_15_r20/external/pytorch/torch/amp/ |
| H A D | autocast_mode.py | 382 return torch.overrides.handle_torch_function(
392 return torch.overrides.handle_torch_function(torch.amp._exit_autocast, [], mode)
|
| /aosp_15_r20/external/pytorch/torch/autograd/ |
| H A D | __init__.py | 16 from torch.overrides import handle_torch_function, has_torch_function, is_tensor_like
445 return handle_torch_function(
|
| /aosp_15_r20/external/pytorch/docs/source/ |
| H A D | torch.overrides.rst | 22 .. autofunction:: handle_torch_function
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_overrides.py | 15 handle_torch_function,
42 return handle_torch_function(foo, (a, b, c), a, b, c=c)
50 return handle_torch_function(bar, (a,), a)
56 return handle_torch_function(baz, (a, b), a, b)
62 return handle_torch_function(quux, (a,), a)
|
| /aosp_15_r20/external/pytorch/torch/csrc/ |
| H A D | Device.cpp | 60 return handle_torch_function( in THPDevice_pynew()
|
| /aosp_15_r20/external/pytorch/torch/_prims/ |
| H A D | __init__.py | 32 from torch.overrides import handle_torch_function, has_torch_function
|