DLPrimitives
Public Member Functions | List of all members
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:
dlprim::ModelBase

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: