ONE - On-device Neural Engine
|
Namespaces | |
namespace | kernels |
namespace | operation |
Data Structures | |
class | Args |
class | ConfigFile |
class | Operation |
class | OperationLoader |
Typedefs | |
using | OperationInfo = std::map< std::string, std::string > |
Functions | |
void | check_valid_key (const std::string &key, OperationInfo &info) |
std::vector< int > | dims (const std::string &src) |
std::vector< int > | get_key_dims (const std::string &key, OperationInfo &info) |
int | get_key_int (const std::string &key, OperationInfo &info) |
std::string | get_key_string (const std::string &key, OperationInfo &info) |
using kbenchmark::OperationInfo = typedef std::map<std::string, std::string> |
Definition at line 62 of file ConfigFile.h.
void kbenchmark::check_valid_key | ( | const std::string & | key, |
OperationInfo & | info | ||
) |
Definition at line 28 of file Utils.h.
References info.
Referenced by get_key_dims(), get_key_int(), and get_key_string().
std::vector< int > kbenchmark::dims | ( | const std::string & | src | ) |
Definition at line 35 of file Utils.h.
Referenced by get_key_dims().
std::vector< int > kbenchmark::get_key_dims | ( | const std::string & | key, |
OperationInfo & | info | ||
) |
Definition at line 50 of file Utils.h.
References check_valid_key(), dims(), and info.
Referenced by kbenchmark::operation::Convolution::params(), and kbenchmark::operation::TransposeConv::params().
int kbenchmark::get_key_int | ( | const std::string & | key, |
OperationInfo & | info | ||
) |
Definition at line 56 of file Utils.h.
References check_valid_key(), and info.
Referenced by kbenchmark::operation::Convolution::params(), and kbenchmark::operation::TransposeConv::params().
std::string kbenchmark::get_key_string | ( | const std::string & | key, |
OperationInfo & | info | ||
) |
Definition at line 62 of file Utils.h.
References check_valid_key(), and info.
Referenced by kbenchmark::operation::Convolution::params(), and kbenchmark::operation::TransposeConv::params().