mirror of https://github.com/alibaba/MNN.git
42 lines
856 B
C++
42 lines
856 B
C++
//
|
|
// Program.hpp
|
|
// MNNConverter
|
|
//
|
|
// Created by MNN on 2019/09/15.
|
|
// Copyright © 2018, Alibaba Group Holding Limited
|
|
//
|
|
|
|
#ifndef Program_hpp
|
|
#define Program_hpp
|
|
#include <fstream>
|
|
#include <map>
|
|
#include <sstream>
|
|
#include <string>
|
|
#include <MNN/expr/Expr.hpp>
|
|
namespace MNN {
|
|
namespace Express {
|
|
|
|
struct Frame;
|
|
class Program {
|
|
public:
|
|
void emit(std::ostream& output);
|
|
void emitPython(std::ostream& output);
|
|
void emitUtils(std::ostream& output);
|
|
static std::shared_ptr<Program> create(const MNN::NetT* net, bool supportExtra);
|
|
std::vector<VARP> outputs() const {
|
|
return mOutputs;
|
|
}
|
|
bool needGenerateCode() const;
|
|
|
|
private:
|
|
Program() {
|
|
}
|
|
std::vector<std::shared_ptr<Frame>> mFrames;
|
|
std::map<int, VARP> mVars;
|
|
std::vector<VARP> mOutputs;
|
|
};
|
|
} // namespace Express
|
|
}; // namespace MNN
|
|
|
|
#endif
|