def METHOD_NAME(self, input_shape, dtype):
    if dtype == "float16":
        return np.random.random(input_shape).astype(dtype)
    if dtype == "uint8":
        return np.random.randint(0, 255, input_shape).astype(dtype)
    if dtype == "int8":
        return np.random.randint(-128, 127, input_shape).astype(dtype)
    raise RuntimeError(f"dtype {dtype} is not valid.")