Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/
H A Dtest_type_promotion.py358 def _get_test_tensor(self, device, dtype, remove_zeros=False): member in TestTypePromotion
391 first = self._get_test_tensor(device, dt1)
392 second = self._get_test_tensor(device, dt2, op == torch.div)
786 t = self._get_test_tensor(device, dtype, not zeros)
789 mask = self._get_test_tensor(device, torch.bool)