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 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 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(), and luci::CircleOutput::index().
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 | ||
) |