Searched refs:_create_tensor (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/eager/ |
| H A D | tensor_test.py | 39 def _create_tensor(value, device=None, dtype=None): function
55 t = _create_tensor(3, dtype=dtypes.int32)
56 self.assertAllEqual(t, _create_tensor(np.array(3)))
86 t = _create_tensor(values)
103 t = _create_tensor(values, dtype=dtypes.float64)
112 t = _create_tensor(n)
165 actual = _create_tensor([[1.0, 2.0], [3.0, 4.0]])
173 t = _create_tensor(3.0)
175 t = _create_tensor(3.0, dtype=dtypes.float64)
179 self.assertFalse(bool(_create_tensor(False)))
[all …]
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_subclass.py | 42 def _create_tensor(self, tensor_cls): member in TestSubclass
48 x = self._create_tensor(tensor_cls).requires_grad_(tensor_requires_grad)
68 x = self._create_tensor(tensor_cls)
84 x = self._create_tensor(tensor_cls)
103 x = self._create_tensor(tensor_cls)
114 x = self._create_tensor(tensor_cls)
119 output = x + self._create_tensor(torch.Tensor)
131 create_fn = partial(self._create_tensor, tensor_cls)
180 create_fn = partial(self._create_tensor, tensor_cls)
198 output = m(self._create_tensor(torch.Tensor))
[all …]
|
| /aosp_15_r20/external/pytorch/test/distributed/fsdp/ |
| H A D | test_shard_utils.py | 23 def _create_tensor(self, *size): member in TestShardUtilsDistributed
31 tensor = self._create_tensor(*size)
51 def _create_tensor(self, *size): member in TestShardUtilsDistributedDTensor
62 tensor = self._create_tensor(*size)
|
| /aosp_15_r20/external/pytorch/test/distributed/_tensor/ |
| H A D | test_dtensor.py | 826 def _create_tensor(self, size): member in TestDTensorPlacementTypes
841 tensor = self._create_tensor(size)
|