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