ONE - On-device Neural Engine
|
Data Structures | |
class | DepthwiseConv2D |
struct | DepthwiseParams |
class | FullyConnected |
struct | FullyConnectedParams |
struct | PaddingValues |
Enumerations | |
enum class | PaddingType : uint8_t { kNone , kSame , kValid } |
enum class | FullyConnectedWeightsFormat : uint8_t { kDefault , kShuffled4x16Int8 } |
enum class | FusedActFunc { UNDEFINED , NONE , RELU , RELU_N1_TO_1 , RELU6 , TANH , SIGN_BIT } |
Functions | |
tflite::RuntimeShape | tflite_shape (const loco::TensorShape &shape) |
tflite::PaddingType | tflite_padding (const PaddingType type) |
tflite::FullyConnectedWeightsFormat | tflite_weights_format (const FullyConnectedWeightsFormat type) |
void | get_act_minmax (const FusedActFunc act, float &act_min, float &act_max) |
|
strong |
|
strong |
Enumerator | |
---|---|
UNDEFINED | |
NONE | |
RELU | |
RELU_N1_TO_1 | |
RELU6 | |
TANH | |
SIGN_BIT |
|
strong |
void luci::compute::get_act_minmax | ( | const FusedActFunc | act, |
float & | act_min, | ||
float & | act_max | ||
) |
Definition at line 26 of file ConvertValues.cpp.
References NONE, RELU, RELU6, RELU_N1_TO_1, and TANH.
Referenced by luci::compute::DepthwiseConv2D::prepare(), and luci::compute::FullyConnected::prepare().
tflite::PaddingType luci::compute::tflite_padding | ( | const PaddingType | type | ) |
Definition at line 38 of file ConvertTypes.cpp.
Referenced by luci::compute::DepthwiseConv2D::compute().
tflite::RuntimeShape luci::compute::tflite_shape | ( | const loco::TensorShape & | shape | ) |
Definition at line 26 of file ConvertTypes.cpp.
References loco::TensorShape::dim(), loco::Dimension::known(), loco::TensorShape::rank(), and loco::Dimension::value().
Referenced by luci::compute::DepthwiseConv2D::compute(), and luci::compute::FullyConnected::compute().
tflite::FullyConnectedWeightsFormat luci::compute::tflite_weights_format | ( | const FullyConnectedWeightsFormat | type | ) |
Definition at line 52 of file ConvertTypes.cpp.
References kDefault, and kShuffled4x16Int8.
Referenced by luci::compute::FullyConnected::compute().