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)
|