Searched defs:tensor_cls (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_subclass.py | 42 def _create_tensor(self, tensor_cls): argument
47 def test_param_invariants(self, tensor_cls, tensor_requires_grad): argument
67 def test_deepcopy(self, tensor_cls, as_param): argument
82 def test_serialization(self, tensor_cls, as_param): argument
102 def test_repr(self, tensor_cls, as_param): argument
113 def test_type_propagation(self, tensor_cls, as_param): argument
130 def test_module_optimization(self, tensor_cls): argument
173 def test_parametrization(self, tensor_cls, leave_parametrized): argument
205 def test_lazy_module(self, tensor_cls): argument
|
| /aosp_15_r20/external/pytorch/torch/multiprocessing/ |
| H A D | reductions.py | 124 tensor_cls, argument
156 tensor_cls, argument
|