| /aosp_15_r20/external/pytorch/tools/autograd/templates/ |
| H A D | python_variable_methods.cpp | 111 if(r.has_torch_function()){ in THPVariable_size()
145 if(r.has_torch_function()){ in THPVariable_stride()
263 if(r.has_torch_function()){ in THPVariable_contiguous()
309 if(r.has_torch_function()){ in THPVariable_copy_()
437 if(r.has_torch_function()){ in THPVariable_cpu()
469 if(r.has_torch_function()){ in THPVariable_nonzero()
492 if(r.has_torch_function()){ in THPVariable_cuda()
515 if (r.has_torch_function()) { in THPVariable_mtia()
538 if (r.has_torch_function()) { in THPVariable_xpu()
561 if (r.has_torch_function()) { in THPVariable_ipu()
[all …]
|
| H A D | python_nn_functions.cpp | 46 if (r.has_torch_function()) { in THPVariable__parse_to()
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | functional.py | 14 has_torch_function,
74 if has_torch_function(tensors):
389 if has_torch_function(operands):
523 if has_torch_function(tensors):
1390 if has_torch_function(tensors):
1426 if has_torch_function(tensors):
1516 if has_torch_function(tensors):
1554 if has_torch_function(tensors):
1600 if has_torch_function(tensors):
2029 if has_torch_function(matrices):
|
| H A D | _lowrank.py | 9 from torch.overrides import handle_torch_function, has_torch_function
143 ) and has_torch_function(tensor_ops):
251 if type(A) is not torch.Tensor and has_torch_function((A,)):
|
| H A D | __init__.py | 785 if overrides.has_torch_function((a, b)):
825 if overrides.has_torch_function((a, b)):
880 if overrides.has_torch_function((b, t, f)):
2034 if type(condition) is not torch.Tensor and overrides.has_torch_function(
|
| H A D | _lobpcg.py | 10 from torch.overrides import handle_torch_function, has_torch_function
509 ) and has_torch_function(tensor_ops):
|
| H A D | _tensor.py | 24 has_torch_function,
37 if has_torch_function(args):
|
| H A D | overrides.py | 243 torch.nn.functional.has_torch_function,
267 has_torch_function,
1578 if has_torch_function(relevant_args):
1754 has_torch_function = _add_docstr( variable
|
| H A D | _ops.py | 423 if torch.overrides.has_torch_function(flat_args):
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | python_torch_functions_manual.cpp | 129 if (r.has_torch_function()) { in THPVariable_as_tensor()
180 if (r.has_torch_function()) { \
231 if (r.has_torch_function()) { in THPVariable_sparse_coo_tensor()
257 if (r.has_torch_function()) { in THPVariable_tensor()
282 if (r.has_torch_function()) { in THPVariable_get_device()
340 if (r.has_torch_function()) { in THPVariable_asarray()
438 if (r.has_torch_function()) { in THPVariable_nonzero()
476 if (r.has_torch_function()) { in THPVariable_numel()
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/variables/ |
| H A D | torch_function.py | 14 from ..utils import get_safe_global_name, has_torch_function, is_tensor_base_attr_getter
235 has_torch_function(arg) for arg in _get_all_args(args, kwargs)
244 [arg for arg in all_args if has_torch_function(arg)],
|
| H A D | torch.py | 34 has_torch_function,
483 torch.overrides.has_torch_function,
494 any(has_torch_function(x) for x in elems),
|
| H A D | user_defined.py | 43 has_torch_function,
651 assert has_torch_function(
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_overrides.py | 16 has_torch_function,
41 if has_torch_function((a, b, c)):
49 if has_torch_function((a,)):
55 if has_torch_function((a, b)):
61 if has_torch_function((a,)):
411 has_torch_function(object())
|
| /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
444 if has_torch_function(overridable_args):
|
| /aosp_15_r20/external/pytorch/docs/source/ |
| H A D | torch.overrides.rst | 24 .. autofunction:: has_torch_function
|
| /aosp_15_r20/external/pytorch/torch/jit/ |
| H A D | _script.pyi | 42 has_torch_function as has_torch_function,
|
| H A D | _script.py | 46 has_torch_function,
1725 _register_builtin(has_torch_function, "aten::has_torch_function")
|
| /aosp_15_r20/external/pytorch/torch/nn/modules/ |
| H A D | transformer.py | 488 if torch.overrides.has_torch_function(tensor_args):
852 if torch.overrides.has_torch_function(tensor_args):
|
| H A D | activation.py | 1286 if torch.overrides.has_torch_function(tensor_args):
|
| /aosp_15_r20/external/pytorch/torch/csrc/ |
| H A D | Device.cpp | 59 if (r.has_torch_function()) { in THPDevice_pynew()
|
| /aosp_15_r20/external/pytorch/torch/csrc/utils/ |
| H A D | python_arg_parser.h | 231 inline bool has_torch_function();
381 inline bool PythonArgs::has_torch_function() { in has_torch_function() function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/ |
| H A D | interned_strings.h | 226 _(aten, has_torch_function) \
|
| /aosp_15_r20/external/pytorch/test/nn/ |
| H A D | test_multihead_attention.py | 710 has_torch_func = torch.overrides.has_torch_function(
|
| /aosp_15_r20/external/pytorch/torch/nn/ |
| H A D | functional.py | 23 has_torch_function,
5983 if has_torch_function(tens_ops):
|