| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_python_dispatch.py | 915 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): member in TestPythonDispatch.test_list_ret.A
935 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): member in TestPythonDispatch.test_invalid_ret.A
1004 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): member in TestPythonDispatch.test_subclass_priority.A
1013 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): member in TestPythonDispatch.test_subclass_priority.B
1109 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): member in TestPythonDispatch.test_new_ones.MyTensor
1117 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): member in TestPythonDispatch.test_like.MyTensor
1191 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): member in TestPythonDispatch.test_make_wrapper_subclass_propagates_metadata.WrapperTensor
1247 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): member in TestPythonDispatch.test_deepcopy_wrapper_subclass_with_clone_returning_different_type.MyWrapperTensor
1318 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): member in TestPythonDispatch.test_wrapper_subclass_extra_dispatch_keys.ExtraKeysTensor
1365 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): member in TestPythonDispatch.test_index_put_where_only_index_is_subclass.MyTensor
[all …]
|
| H A D | test_autocast.py | 216 def __torch_dispatch__(self, func, types, args=(), kwargs=None): member in WeightDTypeCastCounterMode
325 def __torch_dispatch__(self, func, types, args=(), kwargs=None): member in TestAutocastMPS.test_cast_cache_is_global.WeightDTypeCastCounterMode
|
| H A D | test_schema_check.py | 76 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): member in IncorrectAliasTensor
465 def __torch_dispatch__(self, func, types, args=(), kwargs=None): member in TestSchemaCheck.test_schema_info_bind_basic.SchemaInfoBindTestMode
|
| H A D | test_dynamic_shapes.py | 145 def __torch_dispatch__(cls, func_overload, types, args=(), kwargs=None): member in FakeSymbolicTensor
904 def __torch_dispatch__(self, func, types, args=(), kwargs=None): member in TestPySymInt.test_symint_as_scalar.TestSymInt
|
| H A D | test_subclass.py | 245 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): member in TestSubclass.test_non_rewrapping_torch_dispatch_subclass_as_parameter_throws_for_detach.NonRewrappingTensor
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_subclass.py | 70 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): member in DiagTensorBelow
125 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): member in SparseTensor
251 def __torch_dispatch__(cls, func, types, args, kwargs): member in SubclassWithTensorFactory
|
| H A D | two_tensor.py | 53 def __torch_dispatch__(cls, func, types, args, kwargs): member in TwoTensor
82 def __torch_dispatch__(self, func, types, args=(), kwargs=None): member in TwoTensorMode
|
| H A D | logging_tensor.py | 74 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): member in LoggingTensor
87 def __torch_dispatch__(self, func, types, args=(), kwargs=None): member in LoggingTensorMode
|
| H A D | composite_compliance.py | 162 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): member in generate_cct_and_mode.CompositeCompliantTensor
171 def __torch_dispatch__(self, func, types, args=(), kwargs=None): member in generate_cct_and_mode.CompositeCompliantTensorMode
|
| H A D | custom_tensor.py | 45 def __torch_dispatch__(cls, func, types, args, kwargs): member in ConstantExtraMetadataTensor
|
| /aosp_15_r20/external/pytorch/torch/utils/ |
| H A D | _python_dispatch.py | 82 def __torch_dispatch__(self, func, types, args=(), kwargs=None): member in TorchDispatchMode
288 def __torch_dispatch__(self, func, types, args=(), kwargs=None): member in BaseTorchDispatchMode
|
| H A D | checkpoint.py | 1276 def __torch_dispatch__(self, func, types, args=(), kwargs=None): member in _CachingTorchDispatchMode
1307 def __torch_dispatch__(self, func, types, args=(), kwargs=None): member in _CachedTorchDispatchMode
|
| /aosp_15_r20/external/pytorch/test/dynamo/ |
| H A D | test_subclasses.py | 256 def __torch_dispatch__(cls, func, types, args, kwargs=None): member in ScaledTensor
305 def __torch_dispatch__(cls, func, types, args, kwargs=None): member in OptionalScaledTensor
356 def __torch_dispatch__(cls, func, types, args, kwargs): member in CtxSubclassTensor
1229 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): member in SubclassTests.test_wrapper_subclass_guards_on_inner_tensor.DoubleSizeMaybeAddGeThreeTensor
1482 def __torch_dispatch__(cls, func, types, args, kwargs): member in SubclassTests.test_subclass_constructor_proxying.SubclassTensor
1576 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): member in SubclassTests.test_torch_function_subclass_survives_into_aot_autograd.SubTensor
|
| /aosp_15_r20/external/pytorch/torch/_subclasses/ |
| H A D | functional_tensor.py | 182 def __torch_dispatch__(self, func, types, args=(), kwargs=None): member in FunctionalTensor
369 def __torch_dispatch__(self, func, types, args=(), kwargs=None): member in FunctionalTensorMode
|
| H A D | fake_utils.py | 93 def __torch_dispatch__(self, func, types, args=(), kwargs=None): member in CrossRefFakeMode
|
| H A D | schema_check_mode.py | 79 def __torch_dispatch__(self, func, types, args=(), kwargs=None): member in SchemaCheckMode
|
| /aosp_15_r20/external/pytorch/test/export/ |
| H A D | opinfo_schema.py | 74 def __torch_dispatch__(self, func, types, args=(), kwargs=None): member in PreDispatchSchemaCheckMode
|
| /aosp_15_r20/external/pytorch/test/quantization/core/experimental/ |
| H A D | test_bits.py | 21 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): member in Int16Tensor
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | freezing.py | 134 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): member in ErasedTensor
|
| /aosp_15_r20/external/pytorch/torch/distributed/tensor/examples/ |
| H A D | torchrec_sharding_example.py | 86 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): member in LocalShardsWrapper
|
| /aosp_15_r20/external/pytorch/torch/distributed/_tools/ |
| H A D | runtime_estimator.py | 448 … def __torch_dispatch__(self, func, types, args=..., kwargs=None): # type: ignore[no-untyped-def] member in RuntimeEstimator
|
| H A D | memory_tracker.py | 26 def __torch_dispatch__(self, func, types, args=..., kwargs=None): member in MemoryProfileDispatchMode
|
| /aosp_15_r20/external/pytorch/torch/distributed/tensor/ |
| H A D | _shards_wrapper.py | 83 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): member in LocalShardsWrapper
|
| /aosp_15_r20/external/pytorch/test/quantization/core/ |
| H A D | test_utils.py | 212 def __torch_dispatch__(cls, func, types, args, kwargs=None): member in TestUtils.test_uint1_7_dtype.UInt4Tensor
|
| /aosp_15_r20/external/pytorch/test/profiler/ |
| H A D | test_profiler_tree.py | 68 def __torch_dispatch__(cls, func, types, args=(), kwargs=None): member in TorchDispatchTensor
|