hidet.graph.frontend.onnx

hidet.graph.frontend.from_onnx(model)

Load an onnx model to hidet.graph.nn.Module.

Parameters:

model (Union[str, onnx.ModelProto]) – The path or model proto of given onnx model.

Returns:

ret – The loaded model.

Return type:

OnnxModule