Function generate_np_inputs

tests/python/relax/test_frontend_stablehlo.py:41–63

Generate numpy data as the inputs of model Parameters ---------- input_shapes: Union[Tuple, List[Tuple]] shapes for inputs dtype: str the data type of inputs Results ------- out: List[np.ndarray] numpy input data

(
    input_shapes: tuple | list[tuple], dtype: str = "float32"
)

def generate_np_inputs(
    input_shapes: tuple | list[tuple], dtype: str = "float32"
) -> np.ndarray | list[np.ndarray]:
    """Generate numpy data as the inputs of model

    Parameters
    ----------
    input_shapes: Union[Tuple, List[Tuple]]
        shapes for inputs
    dtype: str
        the data type of inputs

    Results
    -------
    out: List[np.ndarray]
        numpy input data
    """
    if not isinstance(input_shapes[0], list | tuple):
        return [np.random.uniform(size=input_shapes).astype(dtype)]
    out = []
    for input_shape in input_shapes:
        out.append(np.random.uniform(size=input_shape).astype(dtype))
    return out


def np2jnp(inputs_np: np.ndarray | list[np.ndarray]):