(self)
| 52 | t = _rand_tensor([4, 8], torch.float32, torch.device("cpu")) |
| 53 | assert t.shape == (4, 8) |
| 54 | assert t.dtype == torch.float32 |
| 55 | |
| 56 | def test_float16(self): |
| 57 | t = _rand_tensor([3], torch.float16, torch.device("cpu")) |
| 58 | assert t.dtype == torch.float16 |
| 59 |
nothing calls this directly
no test coverage detected