Searched refs:float8_type (Results 1 – 1 of 1) sorted by relevance
588 def test_float8_support(self, float8_type): argument591 input = input.to(float8_type)592 return input + torch.tensor(1.0, dtype=float8_type)