/** * \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_