Searched refs:_get_test_tensor (Results 1 – 1 of 1) sorted by relevance
358 def _get_test_tensor(self, device, dtype, remove_zeros=False): member in TestTypePromotion391 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)