Home
last modified time | relevance | path

Searched refs:_create_tensor (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dtensor_test.py39 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 Dtest_subclass.py42 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 Dtest_shard_utils.py23 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 Dtest_dtensor.py826 def _create_tensor(self, size): member in TestDTensorPlacementTypes
841 tensor = self._create_tensor(size)