nndeploy C++ API  0.2.0
nndeploy C++ API
Public Member Functions | Public Attributes | List of all members
nndeploy::ir::ModelDesc Class Reference

参照onnx的格式,描述模型的结构 More...

#include <ir.h>

Public Member Functions

base::Status deserializeStructureFromJson (const std::string &path, const std::vector< ValueDesc > &input)
 
base::Status deserializeStructureFromJson (rapidjson::Value &json, const std::vector< ValueDesc > &input)
 
base::Status deserializeStructureFromJsonStr (const std::string &json_str, const std::vector< ValueDesc > &input)
 
base::Status dump (std::ostream &stream)
 
 ModelDesc ()
 
base::Status serializeStructureToJson (const std::string &path)
 
base::Status serializeStructureToJson (rapidjson::Value &json, rapidjson::Document::AllocatorType &allocator)
 
base::Status serializeStructureToJsonStr (std::string &json_str)
 
virtual ~ModelDesc ()
 

Public Attributes

std::vector< std::shared_ptr< ValueDesc > > inputs_
 
std::unordered_map< std::string, std::string > metadata_
 
std::string name_
 
std::vector< std::shared_ptr< OpDesc > > op_descs_
 
std::vector< std::shared_ptr< ValueDesc > > outputs_
 
std::vector< std::shared_ptr< ValueDesc > > values_
 
std::map< std::string, device::Tensor * > weights_
 

Detailed Description

参照onnx的格式,描述模型的结构

Definition at line 109 of file ir.h.

Constructor & Destructor Documentation

◆ ModelDesc()

nndeploy::ir::ModelDesc::ModelDesc ( )

◆ ~ModelDesc()

virtual nndeploy::ir::ModelDesc::~ModelDesc ( )
virtual

Member Function Documentation

◆ deserializeStructureFromJson() [1/2]

base::Status nndeploy::ir::ModelDesc::deserializeStructureFromJson ( const std::string &  path,
const std::vector< ValueDesc > &  input 
)

◆ deserializeStructureFromJson() [2/2]

base::Status nndeploy::ir::ModelDesc::deserializeStructureFromJson ( rapidjson::Value &  json,
const std::vector< ValueDesc > &  input 
)

◆ deserializeStructureFromJsonStr()

base::Status nndeploy::ir::ModelDesc::deserializeStructureFromJsonStr ( const std::string &  json_str,
const std::vector< ValueDesc > &  input 
)

◆ dump()

base::Status nndeploy::ir::ModelDesc::dump ( std::ostream &  stream)

◆ serializeStructureToJson() [1/2]

base::Status nndeploy::ir::ModelDesc::serializeStructureToJson ( const std::string &  path)

◆ serializeStructureToJson() [2/2]

base::Status nndeploy::ir::ModelDesc::serializeStructureToJson ( rapidjson::Value &  json,
rapidjson::Document::AllocatorType &  allocator 
)

◆ serializeStructureToJsonStr()

base::Status nndeploy::ir::ModelDesc::serializeStructureToJsonStr ( std::string &  json_str)

Member Data Documentation

◆ inputs_

std::vector<std::shared_ptr<ValueDesc> > nndeploy::ir::ModelDesc::inputs_

Definition at line 148 of file ir.h.

◆ metadata_

std::unordered_map<std::string, std::string> nndeploy::ir::ModelDesc::metadata_

Definition at line 146 of file ir.h.

◆ name_

std::string nndeploy::ir::ModelDesc::name_

Definition at line 144 of file ir.h.

◆ op_descs_

std::vector<std::shared_ptr<OpDesc> > nndeploy::ir::ModelDesc::op_descs_

Definition at line 152 of file ir.h.

◆ outputs_

std::vector<std::shared_ptr<ValueDesc> > nndeploy::ir::ModelDesc::outputs_

Definition at line 150 of file ir.h.

◆ values_

std::vector<std::shared_ptr<ValueDesc> > nndeploy::ir::ModelDesc::values_

Definition at line 156 of file ir.h.

◆ weights_

std::map<std::string, device::Tensor *> nndeploy::ir::ModelDesc::weights_

Definition at line 154 of file ir.h.


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