|
ONE - On-device Neural Engine
|
Data Structures | |
| class | CircleEvalDiff |
| class | DirectoryLoader |
| class | HDF5Loader |
| class | InputDataLoader |
| class | MAEPrinter |
| class | MAPEPrinter |
| class | MetricPrinter |
| class | MPEIRPrinter |
| class | MSEPrinter |
| struct | Tensor |
| struct | TensorDataType |
| struct | TensorShape |
| class | TopKMatchPrinter |
Enumerations | |
| enum class | Metric { Undefined , MAE , MAPE , MPEIR , MTOP1 , MTOP5 , MSE } |
| enum class | InputFormat { Undefined , H5 , DIR } |
Functions | |
| std::vector< std::shared_ptr< Tensor > > | interpret (const luci::Module *module, const InputDataLoader::Data &data) |
| void | verifyTypeShape (const luci::CircleInput *input_node, const DataType &dtype, const Shape &shape) |
| std::vector< size_t > | getEachByteSizeOf (const std::vector< loco::Node * > &nodes) |
| size_t | getTotalByteSizeOf (const std::vector< loco::Node * > &nodes) |
| std::unique_ptr< InputDataLoader > | makeDataLoader (const std::string &file_path, const InputFormat &format, const std::vector< loco::Node * > &input_nodes) |
| void | verifyTypeShape (const luci::CircleInput *input_node, const loco::DataType &dtype, const std::vector< loco::Dimension > &shape) |
| INSTANTIATE (loco::DataType::S64) | |
| INSTANTIATE (loco::DataType::S32) | |
| INSTANTIATE (loco::DataType::S16) | |
| INSTANTIATE (loco::DataType::U8) | |
| INSTANTIATE (loco::DataType::FLOAT32) | |
| std::shared_ptr< Tensor > | createEmptyTensor (const luci::CircleNode *node) |
|
strong |
| Enumerator | |
|---|---|
| Undefined | |
| H5 | |
| DIR | |
Definition at line 40 of file InputDataLoader.h.
|
strong |
| Enumerator | |
|---|---|
| Undefined | |
| MAE | |
| MAPE | |
| MPEIR | |
| MTOP1 | |
| MTOP5 | |
| MSE | |
Definition at line 36 of file CircleEvalDiff.h.
| std::shared_ptr< Tensor > circle_eval_diff::createEmptyTensor | ( | const luci::CircleNode * | node | ) |
Definition at line 90 of file Tensor.cpp.
References luci::CircleNode::name().
Referenced by circle_eval_diff::HDF5Loader::get(), circle_eval_diff::DirectoryLoader::get(), and interpret().
| std::vector< size_t > circle_eval_diff::getEachByteSizeOf | ( | const std::vector< loco::Node * > & | nodes | ) |
Definition at line 52 of file InputDataLoader.cpp.
References loco::must_cast(), and luci::size().
Referenced by circle_eval_diff::DirectoryLoader::get().
| size_t circle_eval_diff::getTotalByteSizeOf | ( | const std::vector< loco::Node * > & | nodes | ) |
Definition at line 73 of file InputDataLoader.cpp.
References loco::must_cast(), and luci::size().
Referenced by circle_eval_diff::DirectoryLoader::DirectoryLoader(), and circle_eval_diff::DirectoryLoader::get().
| circle_eval_diff::INSTANTIATE | ( | loco::DataType::FLOAT32 | ) |
| circle_eval_diff::INSTANTIATE | ( | loco::DataType::S16 | ) |
| circle_eval_diff::INSTANTIATE | ( | loco::DataType::S32 | ) |
| circle_eval_diff::INSTANTIATE | ( | loco::DataType::S64 | ) |
| circle_eval_diff::INSTANTIATE | ( | loco::DataType::U8 | ) |
| std::vector< std::shared_ptr< Tensor > > circle_eval_diff::interpret | ( | const luci::Module * | module, |
| const InputDataLoader::Data & | data | ||
| ) |
Definition at line 109 of file CircleEvalDiff.cpp.
References createEmptyTensor(), luci::CircleInput::index(), luci::CircleOutput::index(), and loco::must_cast().
Referenced by circle_eval_diff::CircleEvalDiff::evalDiff().
| std::unique_ptr< InputDataLoader > circle_eval_diff::makeDataLoader | ( | const std::string & | file_path, |
| const InputFormat & | format, | ||
| const std::vector< loco::Node * > & | input_nodes | ||
| ) |
Definition at line 220 of file InputDataLoader.cpp.
Referenced by circle_eval_diff::CircleEvalDiff::evalDiff().
| void circle_eval_diff::verifyTypeShape | ( | const luci::CircleInput * | input_node, |
| const DataType & | dtype, | ||
| const Shape & | shape | ||
| ) |
Definition at line 36 of file InputDataLoader.cpp.
Referenced by circle_eval_diff::HDF5Loader::get().
| void circle_eval_diff::verifyTypeShape | ( | const luci::CircleInput * | input_node, |
| const loco::DataType & | dtype, | ||
| const std::vector< loco::Dimension > & | shape | ||
| ) |