dlprim::ONNXModel Class Reference

External model for loading ONNX models for inference with dlprim. More...

#include <include/dlprim/onnx.hpp>

Inheritance diagram for dlprim::ONNXModel:

Public Member Functions
void	load (std::string const &file_name)
	Parse and prepare the model from ONNX file.
virtual json::value const &	network ()
	Generated network.
virtual Tensor	get_parameter (std::string const &name)
	Query parameter by name, if not found empty/null tensor returned.

Detailed Description

External model for loading ONNX models for inference with dlprim.

Use example:

dlprim::Context ctx(device_name);
model.load(path_to_onnx_file);
dlprim::Net net(ctx);
net.load_model(model);

---

The documentation for this class was generated from the following file:

• include/dlprim/onnx.hpp