22 lines
548 B
C++
22 lines
548 B
C++
/**
|
|
* \brief ONNXIO declarations
|
|
* \author Nan Zhou, nanzhou at kneron dot us
|
|
* \copyright 2019 Kneron Inc. All right reserved.
|
|
*/
|
|
#ifndef PIANO_DYNASTY_INCLUDE_IO_ONNXIO_H_
|
|
#define PIANO_DYNASTY_INCLUDE_IO_ONNXIO_H_
|
|
|
|
#include "onnx/onnx.proto3.pb.h"
|
|
|
|
namespace dynasty {
|
|
namespace io {
|
|
class ONNXIO {
|
|
public:
|
|
static kneron_onnx::ModelProto LoadONNX(const std::string &model_file_path);
|
|
static void SaveONNX(kneron_onnx::ModelProto const &t, const std::string &model_file_path);
|
|
};
|
|
}
|
|
}
|
|
|
|
#endif //PIANO_DYNASTY_INCLUDE_IO_ONNXIO_H_
|