Searched refs:tensor_cls (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_subclass.py | 38 subtest(tensor_cls, name=info.name) for tensor_cls, info in subclass_db.items()])
42 def _create_tensor(self, tensor_cls): argument
43 return subclass_db[tensor_cls].create_fn(3)
47 def test_param_invariants(self, tensor_cls, tensor_requires_grad): argument
48 x = self._create_tensor(tensor_cls).requires_grad_(tensor_requires_grad)
67 def test_deepcopy(self, tensor_cls, as_param): argument
68 x = self._create_tensor(tensor_cls)
75 self.assertIsInstance(x_copy, tensor_cls)
82 def test_serialization(self, tensor_cls, as_param): argument
84 x = self._create_tensor(tensor_cls)
[all …]
|
| /aosp_15_r20/external/pytorch/torch/multiprocessing/ |
| H A D | reductions.py | 124 tensor_cls, argument
145 if tensor_cls == torch.nn.parameter.Parameter:
156 tensor_cls, argument
213 if tensor_cls == torch.nn.parameter.Parameter:
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/variables/ |
| H A D | builtin.py | 2011 tensor_cls = left if isinstance(left, TensorVariable) else right
2016 type(tensor_cls), # handle Ndarrays and Tensors
|