ONE - On-device Neural Engine
|
#include <Interpreter.h>
Public Member Functions | |
Interpreter (const luci::Module *module) | |
Interpreter (const luci::Module *module, IMemoryManager *memory_manager) | |
~Interpreter () | |
void | writeInputTensor (const luci::CircleInput *input_node, const void *data, size_t data_size) |
void | readOutputTensor (const luci::CircleOutput *output_node, void *data, size_t data_size) |
size_t | getOutputTensorSize (const luci::CircleOutput *output_node) |
void | interpret () |
void | attachObserver (ExecutionObserver *observer) |
const Tensor * | getTensor (const loco::Node *node) |
Interpreter (const char *model_data_raw, bool dealloc_input) | |
~Interpreter () | |
void | allocateAndWriteInputTensor (int32_t input_tensor_index, const uint8_t *data, size_t data_size) |
uint8_t * | allocateInputTensor (int32_t input_tensor_index) |
uint8_t * | readOutputTensor (int32_t output_tensor_index) |
int32_t | getInputDataSizeByIndex (int32_t input_tensor_index) |
int32_t | getOutputDataSizeByIndex (int32_t output_tensor_index) |
void | interpret () |
Definition at line 50 of file Interpreter.h.
|
explicit |
Definition at line 73 of file Interpreter.cpp.
References luci_interpreter::ModuleLoader::load().
|
explicit |
Definition at line 86 of file Interpreter.cpp.
References luci_interpreter::ModuleLoader::load().
|
default |
|
explicit |
Definition at line 58 of file Interpreter.cpp.
References luci_interpreter::ModuleLoader::load().
luci_interpreter::Interpreter::~Interpreter | ( | ) |
void luci_interpreter::Interpreter::allocateAndWriteInputTensor | ( | int32_t | input_tensor_index, |
const uint8_t * | data, | ||
size_t | data_size | ||
) |
Definition at line 83 of file Interpreter.cpp.
References flatbuffers::data().
uint8_t * luci_interpreter::Interpreter::allocateInputTensor | ( | int32_t | input_tensor_index | ) |
Definition at line 95 of file Interpreter.cpp.
void luci_interpreter::Interpreter::attachObserver | ( | ExecutionObserver * | observer | ) |
Definition at line 144 of file Interpreter.cpp.
int32_t luci_interpreter::Interpreter::getInputDataSizeByIndex | ( | int32_t | input_tensor_index | ) |
Definition at line 69 of file Interpreter.cpp.
int32_t luci_interpreter::Interpreter::getOutputDataSizeByIndex | ( | int32_t | output_tensor_index | ) |
Definition at line 76 of file Interpreter.cpp.
size_t luci_interpreter::Interpreter::getOutputTensorSize | ( | const luci::CircleOutput * | output_node | ) |
Definition at line 128 of file Interpreter.cpp.
References luci_interpreter::getDataTypeSize(), luci::CircleOutput::index(), luci::CircleNode::name(), and luci::output_node().
|
inline |
Definition at line 69 of file Interpreter.h.
void luci_interpreter::Interpreter::interpret | ( | ) |
Definition at line 142 of file Interpreter.cpp.
Referenced by InterpreterWrapper::interpret(), and record_minmax::RecordMinMax::profileData().
void luci_interpreter::Interpreter::interpret | ( | ) |
void luci_interpreter::Interpreter::readOutputTensor | ( | const luci::CircleOutput * | output_node, |
void * | data, | ||
size_t | data_size | ||
) |
Definition at line 115 of file Interpreter.cpp.
References flatbuffers::data(), luci::CircleOutput::index(), luci::CircleNode::name(), and luci::output_node().
Referenced by InterpreterWrapper::readOutputTensor().
uint8_t * luci_interpreter::Interpreter::readOutputTensor | ( | int32_t | output_tensor_index | ) |
Definition at line 104 of file Interpreter.cpp.
void luci_interpreter::Interpreter::writeInputTensor | ( | const luci::CircleInput * | input_node, |
const void * | data, | ||
size_t | data_size | ||
) |
Definition at line 102 of file Interpreter.cpp.
References flatbuffers::data(), luci::CircleInput::index(), luci::input_node(), and luci::CircleNode::name().
Referenced by record_minmax::RecordMinMax::profileData(), and InterpreterWrapper::writeInputTensor().