def METHOD_NAME(*args, **kwargs):
    if input_type == "float32":
        return np.random.normal(1.0, 6.0, in_shape).astype(np.float32)
    elif input_type == "int32":
        return np.random.normal(1.0, 6.0, in_shape).astype(np.int32)
    elif input_type == "int64":
        return np.random.normal(1.0, 6.0, in_shape).astype(np.int64)