|
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(), and luci::must_cast().
|
default |
|
explicit |
Definition at line 58 of file Interpreter.cpp.
References luci_interpreter::ModuleLoader::load(), and luci::must_cast().
| 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(), and luci::must_cast().
| uint8_t * luci_interpreter::Interpreter::allocateInputTensor | ( | int32_t | input_tensor_index | ) |
Definition at line 95 of file Interpreter.cpp.
References luci::must_cast().
| void luci_interpreter::Interpreter::attachObserver | ( | ExecutionObserver * | observer | ) |
Definition at line 144 of file Interpreter.cpp.
References luci::must_cast().
| int32_t luci_interpreter::Interpreter::getInputDataSizeByIndex | ( | int32_t | input_tensor_index | ) |
Definition at line 69 of file Interpreter.cpp.
References luci::must_cast().
| int32_t luci_interpreter::Interpreter::getOutputDataSizeByIndex | ( | int32_t | output_tensor_index | ) |
Definition at line 76 of file Interpreter.cpp.
References luci::must_cast().
| 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::must_cast(), 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::must_cast(), 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.
References luci::must_cast().
| 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(), luci::must_cast(), and luci::CircleNode::name().
Referenced by record_minmax::RecordMinMax::profileData(), and InterpreterWrapper::writeInputTensor().