| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_overrides.py | 145 def __torch_function__(cls, func, types, args=(), kwargs=None): member in DiagonalTensor
215 def __torch_function__(cls, func, types, args=(), kwargs=None): member in SubTensor
366 def __torch_function__(cls, func, types, args=(), kwargs=None): member in TensorLike
600 def __torch_function__(cls, func, types, args=(), kwargs=None): member in TestTorchFunctionOverride.test_grad.Dummy
612 def __torch_function__(cls, func, types, args=(), kwargs=None): member in TestTorchFunctionOverride.test_pow_rpow.NothingImplemented
617 def __torch_function__(cls, func, types, args=(), kwargs=None): member in TestTorchFunctionOverride.test_pow_rpow.RPowOnly
817 def __torch_function__(cls, func, types, args=(), kwargs=None): member in Wrapper
1015 def __torch_function__(cls, func, types, args, kwargs): member in TestWrapTorchFunction.test_wrap_torch_function.A
1032 def __torch_function__(cls, func, types, args, kwargs=None): member in TestIndexing.test_getitem.A
1042 def __torch_function__(cls, func, types, args, kwargs=None): member in TestIndexing.test_getitem_subclass.A
[all …]
|
| /aosp_15_r20/external/pytorch/test/dynamo/ |
| H A D | test_subclasses.py | 151 def __torch_function__(cls, func, types, args=(), kwargs=None): member in BaseTorchFunction
159 def __torch_function__(cls, func, types, args=(), kwargs=None): member in MockSubclass
170 def __torch_function__(cls, func, types, args=(), kwargs=None): member in AttrSubclass
179 def __torch_function__(cls, func, types, args=(), kwargs=None): member in DummyNDim
194 def __torch_function__(cls, func, types, args=(), kwargs=None): member in WrapperSubclass
206 def __torch_function__(cls, func, types, args=(), kwargs=None): member in SigmoidToExpSubclass
449 def __torch_function__(cls, func, types, args=(), kwargs=None): member in SubclassTests.test_no_call_to_new.BadNewTorchFunction
480 def __torch_function__(cls, func, types, args=(), kwargs=None): member in SubclassTests.test_no_torch_function_recompiles.NJT
585 def __torch_function__(cls, func, types, args=(), kwargs=None): member in SubclassTests.test_return_local_subclass.LocalSubclass
609 def __torch_function__( member in SubclassTests.test_torch_function_list_args.MyClass
[all …]
|
| /aosp_15_r20/external/pytorch/torch/nn/ |
| H A D | parameter.py | 92 __torch_function__ = _disabled_torch_function_impl variable in Parameter
161 def __torch_function__(cls, func, types, args=(), kwargs=None): member in UninitializedTensorMixin
251 __torch_function__ = _disabled_torch_function_impl variable in Buffer
|
| /aosp_15_r20/external/pytorch/functorch/dim/ |
| H A D | wrap_type.py | 29 def _py_wrap_method(orig, __torch_function__): argument
36 def wrap_type(use_c, to_patch, pattern, __torch_function__): argument
|
| H A D | __init__.py | 48 __torch_function__ = classmethod(_C.__torch_function__) variable in _Tensor
51 __torch_function__ = reference.__torch_function__ variable in _Tensor
|
| H A D | reference.py | 167 def __torch_function__(self, orig, cls, args, kwargs=empty_dict): function
|
| /aosp_15_r20/external/pytorch/benchmarks/overrides_benchmark/ |
| H A D | common.py | 21 def __torch_function__(cls, func, types, args=(), kwargs=None): member in WithTorchFunction
30 def __torch_function__(cls, func, types, args=(), kwargs=None): member in SubWithTorchFunction
|
| /aosp_15_r20/external/pytorch/test/nn/ |
| H A D | test_load_state_dict.py | 507 def __torch_function__(cls, func, types, args=(), kwargs=None): member in MyLoadTensor
515 def __torch_function__(cls, func, types, args=(), kwargs=None): member in MyLoadTensor2
521 def __torch_function__(cls, func, types, args=(), kwargs=None): member in MyBrokenLoadTensor
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | overrides.py | 2032 def __torch_function__(self, func, types, args=(), kwargs=None): member in TorchFunctionMode
2080 def __torch_function__(self, func, types, args=(), kwargs=None): member in BaseTorchFunctionMode
|
| /aosp_15_r20/external/pytorch/torch/export/ |
| H A D | _safeguard.py | 18 def __torch_function__(self, func, types, args=(), kwargs=None): member in AutogradStateOpsFailSafeguard
|
| /aosp_15_r20/external/pytorch/torch/_prims/ |
| H A D | context.py | 105 def __torch_function__( member in TorchRefsMode
|
| /aosp_15_r20/external/pytorch/torch/utils/ |
| H A D | _device.py | 102 def __torch_function__(self, func, types, args=(), kwargs=None): member in DeviceContext
|
| /aosp_15_r20/external/pytorch/test/quantization/core/experimental/ |
| H A D | test_bits.py | 44 def __torch_function__(cls, func, types, args=(), kwargs=None): member in Int16Tensor
|
| /aosp_15_r20/external/pytorch/torch/nn/attention/ |
| H A D | bias.py | 282 def __torch_function__(cls, func, types, args=(), kwargs=None): member in CausalBias
|
| /aosp_15_r20/external/pytorch/torch/nn/utils/_expanded_weights/ |
| H A D | expanded_weights_impl.py | 125 def __torch_function__(cls, func, _, args=(), kwargs=None): member in ExpandedWeight
|
| /aosp_15_r20/external/pytorch/test/profiler/ |
| H A D | test_profiler_tree.py | 56 def __torch_function__(cls, func, types, args=(), kwargs=None): member in TorchFunctionTensor
|
| /aosp_15_r20/external/pytorch/torch/_export/ |
| H A D | non_strict_utils.py | 507 def __torch_function__(self, func, types, args=(), kwargs=None): member in _NonStrictTorchFunctionHandler
|
| /aosp_15_r20/external/pytorch/torch/sparse/ |
| H A D | semi_structured.py | 198 __torch_function__ = torch._C._disabled_torch_function_impl variable in SparseSemiStructuredTensor
|
| /aosp_15_r20/external/pytorch/torch/masked/maskedtensor/ |
| H A D | core.py | 285 def __torch_function__(cls, func, types, args=(), kwargs=None): member in MaskedTensor
|
| /aosp_15_r20/external/pytorch/torch/nested/_internal/ |
| H A D | nested_tensor.py | 298 def __torch_function__(cls, func, types, args=(), kwargs=None): member in NestedTensor
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_subclass.py | 168 def __torch_function__(cls, func, types, args=(), kwargs=None): member in NonWrapperTensor
|
| /aosp_15_r20/external/executorch/exir/ |
| H A D | tracer.py | 263 def __torch_function__( member in PythonTensor
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/optests/ |
| H A D | generate_tests.py | 561 def __torch_function__(self, func, types, args=(), kwargs=None): member in OpCheckMode
|
| /aosp_15_r20/external/pytorch/torch/fx/ |
| H A D | proxy.py | 499 def __torch_function__(cls, orig_method, types, args=None, kwargs=None): member in Proxy
|
| /aosp_15_r20/external/pytorch/torch/distributed/_shard/sharded_tensor/ |
| H A D | api.py | 1136 def __torch_function__(cls, func, types, args=(), kwargs=None): member in ShardedTensor
|