9 #ifndef Interpreter_hpp 10 #define Interpreter_hpp 76 const std::string& name()
const;
79 const std::string& type()
const;
90 typedef std::function<bool(
const std::vector<Tensor*>&,
const std::string& )>
TensorCallBack;
101 static Interpreter* createFromFile(
const char* file);
108 static Interpreter* createFromBuffer(
const void* buffer,
size_t size);
124 Session* createMultiPathSession(
const std::vector<ScheduleConfig>& configs);
131 bool releaseSession(Session* session);
138 void resizeSession(Session* session);
151 ErrorCode runSession(Session* session)
const;
162 bool sync =
false)
const;
181 Tensor* getSessionInput(
const Session* session,
const char* name);
188 Tensor* getSessionOutput(
const Session* session,
const char* name);
195 const std::map<std::string, Tensor*>& getSessionOutputAll(
const Session* session)
const;
201 const std::map<std::string, Tensor*>& getSessionInputAll(
const Session* session)
const;
209 void resizeTensor(
Tensor* tensor,
const std::vector<int>& dims);
217 const Backend* getBackend(
const Session* session,
const Tensor* tensor)
const;
223 const char* bizCode()
const;
226 static Interpreter* createFromBufferInternal(Content* net);
228 Content* mNet =
nullptr;
Mode
Definition: Interpreter.hpp:35
MNNForwardType
Definition: MNNForwardType.h:12
std::vector< std::string > saveTensors
Definition: Interpreter.hpp:24
MNNForwardType backupType
Definition: Interpreter.hpp:60
Info * mContent
Definition: Interpreter.hpp:87
MNNForwardType type
Definition: Interpreter.hpp:26
Definition: Interpreter.hpp:71
Definition: Interpreter.hpp:43
Definition: Tensor.hpp:25
Definition: Interpreter.hpp:31
#define MNN_PUBLIC
Definition: MNNDefine.h:53
Definition: Interpreter.hpp:22
Definition: Interpreter.hpp:94
Definition: MNNForwardType.h:13
std::function< bool(const std::vector< Tensor * > &, const OperatorInfo *)> TensorCallBackWithInfo
Definition: Interpreter.hpp:91
BackendConfig * backendConfig
Definition: Interpreter.hpp:63
Definition: AutoTime.hpp:16
int numThread
Definition: Interpreter.hpp:28
Path path
Definition: Interpreter.hpp:57
ErrorCode
Definition: ErrorCode.hpp:13
std::function< bool(const std::vector< Tensor * > &, const std::string &)> TensorCallBack
Definition: Interpreter.hpp:90
std::vector< std::string > inputs
Definition: Interpreter.hpp:32
Mode mode
Definition: Interpreter.hpp:55
std::vector< std::string > outputs
Definition: Interpreter.hpp:33