def METHOD_NAME(self):
    dtypes = [
        torch.bool,
        torch.int8,
        torch.int16,
        torch.int32,
        torch.int64,
        torch.float32,
        torch.float64,
    ]
    # restore_mutation.h contains a mapping from activation operators
    # to whether they allow type conversion. Use this checking to
    # guard the mapping, and if any later change breaks the assumption
    # we need to update the mapping correspondingly.
    for activation, dtype in product(activations, dtypes):
        inp = torch.normal(0, 5, size=(4, 4)).to(dtype)
        try:
            out = activation(inp)
            self.assertEqual(dtype, out.dtype)
        except RuntimeError:
            # Skip the not implemented error
            pass