ONE - On-device Neural Engine
|
Data Structures | |
struct | AddFn |
struct | DivFn |
struct | FloorDivFn |
struct | FloorModFn |
struct | MaximumFn |
struct | MinimumFn |
struct | MulFn |
struct | NdArrayDesc |
struct | SquaredDifferenceFn |
struct | SubFn |
struct | TanhFunctor |
Functions | |
OMStatus | Add (const core::ArithmeticQuantParams ¶ms, const uint32_t flat_size, const int8_t *input1_data, const int8_t *input2_data, int8_t *output_data) |
template<typename T1 , typename T2 , typename T3 > | |
OMStatus | ArgMax (const core::OMRuntimeShape &input1_shape, const T1 *input1_data, const T3 *input2_data, const core::OMRuntimeShape &output_shape, T2 *output_data) |
OMStatus | AveragePool (const core::Pool2DParams ¶ms, const core::OMRuntimeShape &input_shape, const int8_t *input_data, const core::OMRuntimeShape &output_shape, int8_t *output_data) |
OMStatus | ConvPerChannel (const core::ConvQuant ¶ms, const core::OMRuntimeShape &input_shape, const int8_t *input_data, const core::OMRuntimeShape &filter_shape, const int8_t *filter_data, const int32_t *bias_data, const core::OMRuntimeShape &output_shape, int8_t *output_data) |
OMStatus | DepthwiseConvPerChannel (const core::ConvQuant ¶ms, const core::OMRuntimeShape &input_shape, const int8_t *input_data, const core::OMRuntimeShape &filter_shape, const int8_t *filter_data, const int32_t *bias_data, const core::OMRuntimeShape &output_shape, int8_t *output_data) |
template<> | |
OMStatus | FullyConnected< int8_t > (const core::FullyConnectedParams ¶ms, const int8_t *input_data, const core::OMRuntimeShape &filter_shape, const int8_t *filter_data, const int32_t *bias_data, const core::OMRuntimeShape &output_shape, int8_t *output_data) |
template<> | |
OMStatus | FullyConnected (const core::FullyConnectedParams ¶ms, const int16_t *input_data, const core::OMRuntimeShape &filter_shape, const int8_t *filter_data, const int64_t *bias_data, const core::OMRuntimeShape &output_shape, int16_t *output_data) |
OMStatus | MaxPool (const core::Pool2DParams ¶ms, const core::OMRuntimeShape &input_shape, const int8_t *input_data, const core::OMRuntimeShape &output_shape, int8_t *output_data) |
OMStatus | Mul (const core::ArithmeticQuantParams ¶ms, const uint32_t flat_size, const int8_t *input1_data, const int8_t *input2_data, int8_t *output_data) |
template<> | |
OMStatus | ReLUCommon< int8_t > (const int flat_size, const int8_t *input_data, int8_t *output_data, const float alpha, const bool is_relu_6) |
template<> | |
OMStatus | Softmax< int8_t, int8_t > (const core::SoftmaxParams ¶ms, const int8_t *input_data, int8_t *output_data) |
OMStatus | SVDF (const core::SVDFQuantParams ¶ms, const int8_t *input_data, const int8_t *weights_feature_data, const int8_t *weights_time_data, const int32_t *bias_data, int8_t *state_data, int8_t *output_data, const core::OMRuntimeShape &input_shape, const core::OMRuntimeShape &weights_feature_shape, const core::OMRuntimeShape &weights_time_shape, const core::OMRuntimeShape &bias_shape, const core::OMRuntimeShape &output_shape) |
template<typename T > | |
OMStatus | Abs (const core::OMRuntimeShape &shape, const T *input_data, T *output_data) |
int8_t | AddFunc (int8_t x, int8_t y, const core::ArithmeticQuantParams ¶ms) |
template<typename T > | |
OMStatus | Add (const core::BinaryArithmeticBroadcastParams ¶ms, const int flat_size, const T *input1_data, const T *input2_data, T *output_data) |
template<typename T > | |
OMStatus | BroadcastAdd4DSlow (const core::BinaryArithmeticBroadcastParams ¶ms, const core::OMRuntimeShape &input1_shape, const T *input1_data, const core::OMRuntimeShape &input2_shape, const T *input2_data, const core::OMRuntimeShape &output_shape, T *output_data) |
OMStatus | BroadcastAdd4DSlow (const core::ArithmeticQuantParams ¶ms, const core::OMRuntimeShape &input1_shape, const int8_t *input1_data, const core::OMRuntimeShape &input2_shape, const int8_t *input2_data, const core::OMRuntimeShape &output_shape, int8_t *output_data) |
template<typename T > | |
OMStatus | AddN (const size_t flat_size, const size_t num_inputs, const T *const *input_data, T *output_data) |
template<typename T1 , typename T2 , typename T3 , typename Cmp > | |
OMStatus | ArgMinMax (const core::OMRuntimeShape &input1_shape, const T1 *input1_data, const T3 *input2_data, const core::OMRuntimeShape &output_shape, T2 *output_data, const Cmp &cmp) |
template<typename T , typename Fn > | |
OMStatus | ArithmeticOp (const core::BinaryArithmeticBroadcastParams ¶ms, const int flat_size, const T *input1_data, const T *input2_data, T *output_data) |
template<typename T > | |
void | ElementWise (const uint32_t size, const core::ArithmeticQuantParams ¶ms, const T *input1_data, const T *input2_data, T *output_data, T(*binary_func)(T, T, const core::ArithmeticQuantParams &)) |
template<typename T , typename Fn > | |
void | ArithmeticOpScalar (const core::BinaryArithmeticBroadcastParams ¶ms, const int flat_size, const T *input_data, const T scalar_value, T *output_data) |
template<typename T , typename Fn > | |
OMStatus | BroadcastArithmeticOp4DSlow (const core::BinaryArithmeticBroadcastParams ¶ms, const core::OMRuntimeShape &input1_shape, const T *input1_data, const core::OMRuntimeShape &input2_shape, const T *input2_data, const core::OMRuntimeShape &output_shape, T *output_data) |
template<typename T > | |
void | BroadcastInput1 (int size, const core::ArithmeticQuantParams ¶ms, const T *input1_data, const T *input2_data, T *output_data, T(*binary_func)(T, T, const core::ArithmeticQuantParams &)) |
template<typename T > | |
void | BroadcastInput2 (int size, const core::ArithmeticQuantParams ¶ms, const T *input1_data, const T *input2_data, T *output_data, T(*binary_func)(T, T, const core::ArithmeticQuantParams &)) |
template<typename T > | |
void | BroadcastRecursiveDimensions (const core::ArithmeticQuantParams ¶ms, int dimension, size_t *input1_offset_p, size_t *input2_offset_p, size_t *output_offset, size_t *compressed_input1_stride, size_t *compressed_input2_stride, size_t *compressed_output_shape, const T *input1_data, const T *input2_data, T *output_data, T(*binary_func)(T, T, const core::ArithmeticQuantParams &)) |
template<typename T > | |
void | BroadcastBinaryFunction6DSlow (const core::ArithmeticQuantParams ¶ms, const core::OMRuntimeShape &input1_shape, const T *input1_data, const core::OMRuntimeShape &input2_shape, const T *input2_data, const core::OMRuntimeShape &output_shape, T *output_data, T(*binary_func)(T, T, const core::ArithmeticQuantParams &)) |
OMStatus | AveragePool (const core::Pool2DParams ¶ms, const core::OMRuntimeShape &input_shape, const float *input_data, const core::OMRuntimeShape &output_shape, float *output_data) |
template<typename T > | |
OMStatus | BatchToSpaceND (const core::OMRuntimeShape &unextended_input1_shape, const T *input1_data, const core::OMRuntimeShape &unextended_input2_shape, const int32_t *block_shape_data, const core::OMRuntimeShape &unextended_input3_shape, const int32_t *crops_data, const core::OMRuntimeShape &unextended_output_shape, T *output_data) |
template<typename T , typename Fn > | |
OMStatus | BinaryOp (const int flat_size, const T *input1_data, const T *input2_data, T *output_data) |
template<typename T , typename Fn > | |
OMStatus | BroadcastBinaryOp4DSlow (const core::OMRuntimeShape &input1_shape, const float *input1_data, const core::OMRuntimeShape &input2_shape, const float *input2_data, const core::OMRuntimeShape &output_shape, float *output_data) |
template<typename FromT , typename ToT > | |
OMStatus | Cast (const core::OMRuntimeShape &input_shape, const FromT *input_data, const core::OMRuntimeShape &output_shape, ToT *output_data) |
template<typename T > | |
OMStatus | Ceil (const core::OMRuntimeShape &input_shape, const T *input_data, const core::OMRuntimeShape &output_shape, T *output_data) |
template<typename T > | |
bool | LessFn (T lhs, T rhs) |
template<typename T > | |
bool | LessEqualFn (T lhs, T rhs) |
template<typename T > | |
bool | EqualFn (T lhs, T rhs) |
template<typename T > | |
bool | GreaterFn (T lhs, T rhs) |
template<typename T > | |
bool | GreaterEqualFn (T lhs, T rhs) |
template<typename T > | |
bool | NotEqualFn (T lhs, T rhs) |
template<typename T > | |
void | ComparisonNoScaling (const int64_t flat_size, const T *input1_data, const T *input2_data, bool *output_data, bool F(T, T)) |
template<typename T , typename AccType > | |
void | BroadcastComparison4DSlowWithScaling (const core::ComparisonParams &op_params, const core::OMRuntimeShape &unextended_input1_shape, const T *input1_data, const core::OMRuntimeShape &unextended_input2_shape, const T *input2_data, const core::OMRuntimeShape &unextended_output_shape, bool *output_data, bool F(AccType, AccType)) |
template<typename T , typename AccType > | |
void | ComparisonWithScaling (const core::ComparisonParams &op_params, const int64_t flat_size, const T *input1_data, const T *input2_data, bool *output_data, bool F(AccType, AccType)) |
template<typename T > | |
void | BroadcastComparison4DSlowNoScaling (const core::ComparisonParams &op_params, const core::OMRuntimeShape &unextended_input1_shape, const T *input1_data, const core::OMRuntimeShape &unextended_input2_shape, const T *input2_data, const core::OMRuntimeShape &unextended_output_shape, bool *output_data, bool F(T, T)) |
template<typename Scalar > | |
OMStatus | Concatenation (core::ConcatenationParams ¶ms, std::vector< uint32_t > &input_shapes, std::vector< const Scalar * > &input_data, const core::OMRuntimeShape &output_shape, Scalar *output_data) |
OMStatus | ConvFloat (const core::FloatConv2D *params, const core::OMRuntimeShape &input_shape, const float *input_data, const core::OMRuntimeShape &filter_shape, const float *filter_data, const float *bias_data, const core::OMRuntimeShape &output_shape, float *output_data) |
template<typename T > | |
OMStatus | Cos (const core::OMRuntimeShape &input_shape, const T *input_data, const core::OMRuntimeShape &output_shape, T *output_data) |
template<typename T > | |
OMStatus | DepthwiseConv2D (const core::FloatConv2D *params, const core::OMRuntimeShape &input_shape, const T *input_data, const core::OMRuntimeShape &filter_shape, const T *filter_data, const T *bias_data, const core::OMRuntimeShape &output_shape, T *output_data) |
template<> | |
OMStatus | DepthwiseConv2D< float > (const core::FloatConv2D *params, const core::OMRuntimeShape &input_shape, const float *input_data, const core::OMRuntimeShape &filter_shape, const float *filter_data, const float *bias_data, const core::OMRuntimeShape &output_shape, float *output_data) |
template<typename InputT , typename OutputT > | |
OMStatus | Dequantize (const core::QuantizationParams op_params, const uint32_t flat_size, const InputT *input_data, OutputT *output_data) |
template<typename T > | |
OMStatus | Div (const core::BinaryArithmeticBroadcastParams ¶ms, const int flat_size, const T *input1_data, const T *input2_data, T *output_data) |
template<typename T > | |
OMStatus | BroadcastDiv4DSlow (const core::BinaryArithmeticBroadcastParams ¶ms, const core::OMRuntimeShape &input1_shape, const T *input1_data, const core::OMRuntimeShape &input2_shape, const T *input2_data, const core::OMRuntimeShape &output_shape, T *output_data) |
OMStatus | Elu (const int flat_size, const float *input_data, float *output_data) |
template<typename T > | |
OMStatus | Exp (const core::OMRuntimeShape &input_shape, const T *input_data, const core::OMRuntimeShape &output_shape, T *output_data) |
template<typename T > | |
OMStatus | Fill (const T *input_data, const core::OMRuntimeShape &output_shape, T *output_data) |
template<typename T > | |
OMStatus | Floor (const core::OMRuntimeShape &input_shape, const T *input_data, const core::OMRuntimeShape &output_shape, T *output_data) |
void | FloorDiv (const int flat_size, const float *input1_data, const float *input2_data, float *output_data) |
void | BroadcastFloorDiv4DSlow (const core::OMRuntimeShape &input1_shape, const float *input1_data, const core::OMRuntimeShape &input2_shape, const float *input2_data, const core::OMRuntimeShape &output_shape, float *output_data) |
void | FloorMod (const int flat_size, const float *input1_data, const float *input2_data, float *output_data) |
void | BroadcastFloorMod4DSlow (const core::OMRuntimeShape &input1_shape, const float *input1_data, const core::OMRuntimeShape &input2_shape, const float *input2_data, const core::OMRuntimeShape &output_shape, float *output_data) |
template<typename InputType , typename WeightType , typename OutputType , typename BiasType > | |
OMStatus | FullyConnected (const core::FullyConnectedParams ¶ms, const InputType *input_data, const core::OMRuntimeShape &filter_shape, const WeightType *filter_data, const BiasType *bias_data, const core::OMRuntimeShape &output_shape, OutputType *output_data) |
template<typename WeightType > | |
OMStatus | FullyConnected (const core::FullyConnectedParams ¶ms, const float *input_data, const core::OMRuntimeShape &filter_shape, const WeightType *filter_data, const float *bias_data, const core::OMRuntimeShape &output_shape, float *output_data) |
template<typename ParamsT , typename IndicesT > | |
OMStatus | GatherND (core::OMRuntimeShape params_shape, const ParamsT *param_data, core::OMRuntimeShape indices_shape, const IndicesT *index_data, ParamsT *output_data) |
OMStatus | GRU (const float *input_data, const float *weight_input_data, const float *weight_hidden_data, const float *bias_input_data, const float *bias_hidden_data, const float *hidden_state_data, float *output_data, float *output_input_data, float *output_hidden_data, const core::OMRuntimeShape &input_shape, const core::OMRuntimeShape &output_shape, const core::OMRuntimeShape &weight_input_shape, const core::OMRuntimeShape &weight_hidden_shape, const size_t intermediate_buffer_size, float *intermediate_buffer) |
OMStatus | L2Normalization (const core::L2NormalizationParams ¶ms, const float *input_data, float *output_data) |
OMStatus | L2Pool (const core::Pool2DParams ¶ms, const core::OMRuntimeShape &input_shape, const float *input_data, const core::OMRuntimeShape &output_shape, float *output_data) |
template<typename T > | |
OMStatus | Log (const core::OMRuntimeShape &input_shape, const T *input_data, const core::OMRuntimeShape &output_shape, T *output_data) |
OMStatus | Logistic (const int flat_size, const float *input_data, float *output_data) |
OMStatus | Logistic (const int flat_size, const int8_t *input_data, float input_scale, int input_zero_point, int8_t *output_data, float output_scale, int output_zero_point) |
OMStatus | LogSoftmax (const core::LogSoftmaxParams ¶ms, const float *input_data, float *output_data) |
OMStatus | Maximum (const int flat_size, const float *input1_data, const float *input2_data, float *output_data) |
OMStatus | BroadcastMaximum4DSlow (const core::OMRuntimeShape &input1_shape, const float *input1_data, const core::OMRuntimeShape &input2_shape, const float *input2_data, const core::OMRuntimeShape &output_shape, float *output_data) |
OMStatus | MaxPool (const core::Pool2DParams ¶ms, const core::OMRuntimeShape &input_shape, const float *input_data, const core::OMRuntimeShape &output_shape, float *output_data) |
OMStatus | Minimum (const int flat_size, const float *input1_data, const float *input2_data, float *output_data) |
template<typename T > | |
OMStatus | BroadcastMinimum4DSlow (const core::OMRuntimeShape &input1_shape, const T *input1_data, const core::OMRuntimeShape &input2_shape, const T *input2_data, const core::OMRuntimeShape &output_shape, T *output_data) |
template<typename T > | |
OMStatus | Mul (const core::BinaryArithmeticBroadcastParams ¶ms, const int flat_size, const T *input1_data, const T *input2_data, T *output_data) |
template<typename T > | |
OMStatus | BroadcastMul4DSlow (const core::BinaryArithmeticBroadcastParams ¶ms, const core::OMRuntimeShape &input1_shape, const T *input1_data, const core::OMRuntimeShape &input2_shape, const T *input2_data, const core::OMRuntimeShape &output_shape, T *output_data) |
template<typename T > | |
OMStatus | BroadcastMul6DSlow (const core::ArithmeticQuantParams ¶ms, const core::OMRuntimeShape &input1_shape, const T *input1_data, const core::OMRuntimeShape &input2_shape, const T *input2_data, const core::OMRuntimeShape &output_shape, T *output_data) |
template<typename T > | |
OMStatus | Neg (const core::OMRuntimeShape &input_shape, const T *input_data, const core::OMRuntimeShape &output_shape, T *output_data) |
OMStatus | Pad (const core::PadParams &op_params, const core::OMRuntimeShape &input_shape, const float *input_data, const float pad_value, const core::OMRuntimeShape &output_shape, float *output_data) |
template<typename InputT , typename OutputT > | |
OMStatus | Quantize (const core::QuantizationParams op_params, const uint32_t flat_size, const InputT *input_data, OutputT *output_data) |
template<typename Type > | |
OMStatus | ReLUCommon (const int flat_size, const Type *input_data, Type *output_data, const float alpha, const bool is_relu_6) |
template<typename T > | |
OMStatus | Round (const core::OMRuntimeShape &input_shape, const T *input_data, const core::OMRuntimeShape &output_shape, T *output_data) |
template<> | |
OMStatus | Round< float > (const core::OMRuntimeShape &input_shape, const float *input_data, const core::OMRuntimeShape &output_shape, float *output_data) |
template<typename T > | |
OMStatus | Rsqrt (const core::OMRuntimeShape &input_shape, const T *input_data, const core::OMRuntimeShape &output_shape, T *output_data) |
template<typename T > | |
OMStatus | Sin (const core::OMRuntimeShape &input_shape, const T *input_data, const core::OMRuntimeShape &output_shape, T *output_data) |
template<typename T > | |
OMStatus | SISOOperation (const core::OMRuntimeShape &input_shape, const T *input_data, const core::OMRuntimeShape &output_shape, T *output_data, std::function< T(T)> const &func) |
template<typename T > | |
OMStatus | Slice (const core::SliceParams &op_params, const core::OMRuntimeShape &input_shape, const T *input_data, T *output_data) |
template<typename T , typename U > | |
OMStatus | Softmax (const core::SoftmaxParams ¶ms, const T *input_data, U *output_data) |
template<> | |
OMStatus | Softmax< float, float > (const core::SoftmaxParams ¶ms, const float *input_data, float *output_data) |
template<typename T > | |
OMStatus | SpaceToBatchND (const core::OMRuntimeShape &unextended_input1_shape, const T *input1_data, const core::OMRuntimeShape &unextended_input2_shape, const int32_t *block_shape_data, const core::OMRuntimeShape &unextended_input3_shape, const int32_t *paddings_data, const core::OMRuntimeShape &unextended_output_shape, T *output_data) |
template<typename T > | |
OMStatus | SpaceToDepth (const int32_t block_size, const core::OMRuntimeShape &unextended_input_shape, const T *input_data, const core::OMRuntimeShape &unextended_output_shape, T *output_data) |
template<typename T > | |
OMStatus | Split (const core::SplitParams ¶ms, const core::OMRuntimeShape &input_shape, const T *input_data, const core::OMRuntimeShape &output_shape, int32_t axis_value) |
template<typename T > | |
OMStatus | Sqrt (const core::OMRuntimeShape &input_shape, const T *input_data, const core::OMRuntimeShape &output_shape, T *output_data) |
template<typename T > | |
OMStatus | Square (const core::OMRuntimeShape &input_shape, const T *input_data, const core::OMRuntimeShape &output_shape, T *output_data) |
template<typename T > | |
OMStatus | SquaredDifference (const core::BinaryArithmeticBroadcastParams ¶ms, const int flat_size, const T *input1_data, const T *input2_data, T *output_data) |
template<typename T > | |
OMStatus | BroadcastSquaredDifference4DSlow (const core::BinaryArithmeticBroadcastParams ¶ms, const core::OMRuntimeShape &input1_shape, const T *input1_data, const core::OMRuntimeShape &input2_shape, const T *input2_data, const core::OMRuntimeShape &output_shape, T *output_data) |
template<typename T > | |
OMStatus | StridedSlice (core::StridedSliceParams &op_params, const core::OMRuntimeShape &unextended_input_shape, const T *input_data, T *output_data) |
int8_t | SubFunc (int8_t x, int8_t y, const core::ArithmeticQuantParams ¶ms) |
template<typename T > | |
OMStatus | Sub (const core::BinaryArithmeticBroadcastParams ¶ms, const int flat_size, const T *input1_data, const T *input2_data, T *output_data) |
template<typename T > | |
OMStatus | BroadcastSub4DSlow (const core::BinaryArithmeticBroadcastParams ¶ms, const core::OMRuntimeShape &input1_shape, const T *input1_data, const core::OMRuntimeShape &input2_shape, const T *input2_data, const core::OMRuntimeShape &output_shape, T *output_data) |
OMStatus | BroadcastSub4DSlow (const core::ArithmeticQuantParams ¶ms, const core::OMRuntimeShape &input1_shape, const int8_t *input1_data, const core::OMRuntimeShape &input2_shape, const int8_t *input2_data, const core::OMRuntimeShape &output_shape, int8_t *output_data) |
OMStatus | SVDF (const float *input_data, const float *weights_feature_data, const float *weights_time_data, const float *bias_data, float *state_data, float *scratch_data, float *output_data, const int rank, const int input_size, const int batch_size, const int num_filters, const int num_units, const int memory_size, const circle::ActivationFunctionType activation) |
template<typename T > | |
OMStatus | Tanh (const core::OMRuntimeShape &input_shape, const T *input_data, const core::OMRuntimeShape &output_shape, T *output_data) |
template<typename T , int N> | |
OMStatus | TransposeImpl (const core::TransposeParams ¶ms, const core::OMRuntimeShape &unextended_input_shape, const T *input_data, const core::OMRuntimeShape &unextended_output_shape, T *output_data) |
template<typename T , int N = 5> | |
OMStatus | Transpose (const core::TransposeParams ¶ms, const core::OMRuntimeShape &unextended_input_shape, const T *input_data, const core::OMRuntimeShape &unextended_output_shape, T *output_data) |
template<typename T > | |
OMStatus | TransposeConv (const core::FloatConv2D *params, const core::OMRuntimeShape &input_shape, const T *input_data, const core::OMRuntimeShape &filter_shape, const T *filter_data, const T *bias_data, const core::OMRuntimeShape &output_shape, T *output_data) |
template<> | |
OMStatus | TransposeConv< float > (const core::FloatConv2D *params, const core::OMRuntimeShape &input_shape, const float *input_data, const core::OMRuntimeShape &filter_shape, const float *filter_data, const float *bias_data, const core::OMRuntimeShape &output_shape, float *output_data) |
template<typename T , typename Fn > | |
OMStatus | UnaryOp (const core::OMRuntimeShape &input_shape, const T *input_data, const core::OMRuntimeShape &output_shape, T *output_data) |
template<typename T > | |
OMStatus | Unpack (const core::SplitParams ¶ms, const core::OMRuntimeShape &input_shape, const T *input_data, const core::OMRuntimeShape &output_shape, int32_t axis_value) |
std::pair< uint32_t, uint32_t > | getUpLowerWeightTensorDepth (core::OpTrainableRankType rank, const uint32_t output_depth) |
std::int32_t | saturatingRoundingDoublingHighMul (std::int32_t a, std::int32_t b) |
int32_t | roundingDivideByPOT (int32_t x, int32_t exponent) |
int32_t | multiplyByQuantizedMultiplier (int32_t x, int32_t quantized_multiplier, int shift) |
int32_t | multiplyByQuantizedMultiplierSmallerThanOneExp (int32_t x, int32_t quantized_multiplier, int left_shift) |
template<typename P > | |
void | getActivationParams (const P ¶ms, int32_t *min, int32_t *max) |
template<typename P > | |
void | getActivationParams (const P ¶ms, float *min, float *max) |
template<typename P > | |
void | getActivationParams (const P ¶ms, int64_t *min, int64_t *max) |
size_t | reducedOutputOffset (const int32_t num_dims, const int32_t *dims, const int32_t *index, const int32_t num_axis, const int32_t *axis) |
bool | nextIndex (const int32_t num_dims, const int32_t *dims, int32_t *current) |
int | MatchingDim (const core::OMRuntimeShape &shape1, int index1, const core::OMRuntimeShape &shape2, int index2) |
int | flatSizeSkipDim (const int32_t *dims_data, int skip_dim, int num_dims) |
int | offset (const int32_t *dims_data, int i0, int i1, int i2, int i3) |
int | offset (const int32_t *dims_data, int i0, int i1, int i2, int i3, int i4) |
template<typename T > | |
T | activationFunctionWithMinMax (T x, T output_activation_min, T output_activation_max) |
template<int MAX_DIM = 6> | |
bool | ReduceDimensionsForBroadcast (const core::OMRuntimeShape &input1_shape, const core::OMRuntimeShape &input2_shape, size_t *compressed_input1_stride, size_t *compressed_input2_stride, size_t *compressed_output_shape) |
template<int N> | |
void | copyDimsToDesc (const core::OMRuntimeShape &input_shape, NdArrayDesc< N > *desc_out) |
template<int N, int DIM, typename Calc > | |
std::enable_if< DIM==N-1, void >::type | NDOpsHelperImpl (const NdArrayDesc< N > &output, const Calc &calc, int indexes[N]) |
template<int N, int DIM, typename Calc > | |
std::enable_if< DIM!=N-1, void >::type | NDOpsHelperImpl (const NdArrayDesc< N > &output, const Calc &calc, int indexes[N]) |
template<int N, typename Calc > | |
void | NDOpsHelper (const NdArrayDesc< N > &output, const Calc &calc) |
template<int N> | |
void | NdArrayDescsForElementwiseBroadcast (const core::OMRuntimeShape &input0_shape, const core::OMRuntimeShape &input1_shape, NdArrayDesc< N > *desc0_out, NdArrayDesc< N > *desc1_out) |
int | subscriptToIndex (const NdArrayDesc< 4 > &desc, int i0, int i1, int i2, int i3) |
int | subscriptToIndex (const NdArrayDesc< 5 > &desc, int indexes[5]) |
bool | processBroadcastShapes (const core::OMRuntimeShape &shape0, const core::OMRuntimeShape &shape1, core::BinaryArithmeticBroadcastParams *params) |
template<> | |
OMStatus | AddN< int8_t > (const size_t flat_size, const size_t num_inputs, const int8_t *const *input_data, int8_t *output_data) |
template<> | |
OMStatus | AddN< int16_t > (const size_t flat_size, const size_t num_inputs, const int16_t *const *input_data, int16_t *output_data) |
template<typename T1 , typename T2 , typename T3 > | |
OMStatus | ArgMin (const core::OMRuntimeShape &input1_shape, const T1 *input1_data, const T3 *input2_data, const core::OMRuntimeShape &output_shape, T2 *output_data) |
template<typename InputType , typename OutputType > | |
OMStatus | Mul (const core::ArithmeticQuantParams ¶ms, uint32_t size, const InputType *input1_data, const InputType *input2_data, OutputType *output_data) |
OMStatus | Sub (const core::ArithmeticQuantParams ¶ms, const uint32_t flat_size, const int8_t *input1_data, const int8_t *input2_data, int8_t *output_data) |
Variables | |
constexpr int | MAX_INDICES_ND = 5 |
|
inline |
Definition at line 33 of file PALAbs.h.
References onert_micro::core::OMRuntimeShape::flatSize(), onert_micro::Ok, and onert_micro::UnknownError.
|
inline |
Definition at line 231 of file PALUtils.h.
Referenced by DepthwiseConv2D< float >(), L2Pool(), and TransposeConv< float >().
OMStatus onert_micro::execute::pal::Add | ( | const core::ArithmeticQuantParams & | params, |
const uint32_t | flat_size, | ||
const int8_t * | input1_data, | ||
const int8_t * | input2_data, | ||
int8_t * | output_data | ||
) |
Definition at line 33 of file PALAdd.h.
References onert_micro::core::ArithmeticQuantParams::input1_multiplier, onert_micro::core::ArithmeticQuantParams::input1_offset, onert_micro::core::ArithmeticQuantParams::input1_shift, onert_micro::core::ArithmeticQuantParams::input2_multiplier, onert_micro::core::ArithmeticQuantParams::input2_offset, onert_micro::core::ArithmeticQuantParams::input2_shift, onert_micro::core::ArithmeticQuantParams::left_shift, onert_micro::Ok, onert_micro::core::ArithmeticQuantParams::output_multiplier, onert_micro::core::ArithmeticQuantParams::output_offset, onert_micro::core::ArithmeticQuantParams::output_shift, onert_micro::core::ArithmeticQuantParams::quantized_activation_max, onert_micro::core::ArithmeticQuantParams::quantized_activation_min, and onert_micro::UnknownError.
OMStatus onert_micro::execute::pal::Add | ( | const core::BinaryArithmeticBroadcastParams & | params, |
const int | flat_size, | ||
const T * | input1_data, | ||
const T * | input2_data, | ||
T * | output_data | ||
) |
Definition at line 50 of file PALAddCommon.h.
References onert_micro::Ok.
int8_t onert_micro::execute::pal::AddFunc | ( | int8_t | x, |
int8_t | y, | ||
const core::ArithmeticQuantParams & | params | ||
) |
Definition at line 30 of file PALAddCommon.h.
References onert_micro::core::ArithmeticQuantParams::input1_multiplier, onert_micro::core::ArithmeticQuantParams::input1_offset, onert_micro::core::ArithmeticQuantParams::input1_shift, onert_micro::core::ArithmeticQuantParams::input2_multiplier, onert_micro::core::ArithmeticQuantParams::input2_offset, onert_micro::core::ArithmeticQuantParams::input2_shift, onert_micro::core::ArithmeticQuantParams::left_shift, multiplyByQuantizedMultiplierSmallerThanOneExp(), onert_micro::core::ArithmeticQuantParams::output_multiplier, onert_micro::core::ArithmeticQuantParams::output_offset, onert_micro::core::ArithmeticQuantParams::output_shift, onert_micro::core::ArithmeticQuantParams::quantized_activation_max, and onert_micro::core::ArithmeticQuantParams::quantized_activation_min.
Referenced by BroadcastAdd4DSlow().
OMStatus onert_micro::execute::pal::AddN | ( | const size_t | flat_size, |
const size_t | num_inputs, | ||
const T *const * | input_data, | ||
T * | output_data | ||
) |
Definition at line 32 of file PALAddNCommon.h.
References onert_micro::Ok.
OMStatus onert_micro::execute::pal::AddN< int16_t > | ( | const size_t | flat_size, |
const size_t | num_inputs, | ||
const int16_t *const * | input_data, | ||
int16_t * | output_data | ||
) |
Definition at line 38 of file PALAddN.h.
References onert_micro::UnsupportedOp.
OMStatus onert_micro::execute::pal::AddN< int8_t > | ( | const size_t | flat_size, |
const size_t | num_inputs, | ||
const int8_t *const * | input_data, | ||
int8_t * | output_data | ||
) |
Definition at line 30 of file PALAddN.h.
References onert_micro::UnsupportedOp.
OMStatus onert_micro::execute::pal::ArgMax | ( | const core::OMRuntimeShape & | input1_shape, |
const T1 * | input1_data, | ||
const T3 * | input2_data, | ||
const core::OMRuntimeShape & | output_shape, | ||
T2 * | output_data | ||
) |
Definition at line 30 of file PALArgMax.h.
References ArgMinMax(), and output_shape.
OMStatus onert_micro::execute::pal::ArgMin | ( | const core::OMRuntimeShape & | input1_shape, |
const T1 * | input1_data, | ||
const T3 * | input2_data, | ||
const core::OMRuntimeShape & | output_shape, | ||
T2 * | output_data | ||
) |
Definition at line 30 of file PALArgMin.h.
References ArgMinMax(), and output_shape.
OMStatus onert_micro::execute::pal::ArgMinMax | ( | const core::OMRuntimeShape & | input1_shape, |
const T1 * | input1_data, | ||
const T3 * | input2_data, | ||
const core::OMRuntimeShape & | output_shape, | ||
T2 * | output_data, | ||
const Cmp & | cmp | ||
) |
Definition at line 32 of file PALArgMinMaxCommon.h.
References onert_micro::core::OMRuntimeShape::dimensionsCount(), onert_micro::core::OMRuntimeShape::dims(), and onert_micro::Ok.
OMStatus onert_micro::execute::pal::ArithmeticOp | ( | const core::BinaryArithmeticBroadcastParams & | params, |
const int | flat_size, | ||
const T * | input1_data, | ||
const T * | input2_data, | ||
T * | output_data | ||
) |
Definition at line 54 of file PALArithmeticOpCommon.h.
References getActivationParams(), and onert_micro::Ok.
|
inline |
Definition at line 80 of file PALArithmeticOpCommon.h.
References getActivationParams().
OMStatus onert_micro::execute::pal::AveragePool | ( | const core::Pool2DParams & | params, |
const core::OMRuntimeShape & | input_shape, | ||
const float * | input_data, | ||
const core::OMRuntimeShape & | output_shape, | ||
float * | output_data | ||
) |
Definition at line 33 of file PALAveragePool2DCommon.h.
References onert_micro::core::Pool2DParams::activation_max, onert_micro::core::Pool2DParams::activation_min, luci_interpreter::RuntimeShape::dims(), onert_micro::core::OMRuntimeShape::dims(), onert_micro::core::Pool2DParams::filter_h, onert_micro::core::Pool2DParams::filter_w, onert_micro::Ok, output_shape, onert_micro::core::Pool2DParams::pad_h, onert_micro::core::Pool2DParams::pad_w, onert_micro::core::Pool2DParams::stride_h, and onert_micro::core::Pool2DParams::stride_w.
OMStatus onert_micro::execute::pal::AveragePool | ( | const core::Pool2DParams & | params, |
const core::OMRuntimeShape & | input_shape, | ||
const int8_t * | input_data, | ||
const core::OMRuntimeShape & | output_shape, | ||
int8_t * | output_data | ||
) |
Definition at line 32 of file PALAveragePool2D.h.
References onert_micro::CmsisNNError, luci_interpreter::RuntimeShape::dims(), onert_micro::core::OMRuntimeShape::dims(), onert_micro::core::Pool2DParams::filter_h, onert_micro::core::Pool2DParams::filter_w, onert_micro::Ok, output_shape, onert_micro::core::Pool2DParams::pad_h, onert_micro::core::Pool2DParams::pad_w, onert_micro::core::Pool2DParams::quantized_activation_max, onert_micro::core::Pool2DParams::quantized_activation_min, onert_micro::core::Pool2DParams::stride_h, and onert_micro::core::Pool2DParams::stride_w.
|
inline |
Definition at line 51 of file PALBatchToSpaceNDCommon.h.
References onert_micro::core::OMRuntimeShape::dimensionsCount(), luci_interpreter::RuntimeShape::dims(), onert_micro::core::OMRuntimeShape::dims(), luci_interpreter::RuntimeShape::dimsData(), onert_micro::core::OMRuntimeShape::dimsData(), offset(), onert_micro::Ok, and output_shape.
|
inline |
Definition at line 62 of file PALBinaryOpCommon.h.
References onert_micro::Ok.
OMStatus onert_micro::execute::pal::BroadcastAdd4DSlow | ( | const core::ArithmeticQuantParams & | params, |
const core::OMRuntimeShape & | input1_shape, | ||
const int8_t * | input1_data, | ||
const core::OMRuntimeShape & | input2_shape, | ||
const int8_t * | input2_data, | ||
const core::OMRuntimeShape & | output_shape, | ||
int8_t * | output_data | ||
) |
Definition at line 68 of file PALAddCommon.h.
References AddFunc(), BroadcastBinaryFunction6DSlow(), onert_micro::Ok, and output_shape.
OMStatus onert_micro::execute::pal::BroadcastAdd4DSlow | ( | const core::BinaryArithmeticBroadcastParams & | params, |
const core::OMRuntimeShape & | input1_shape, | ||
const T * | input1_data, | ||
const core::OMRuntimeShape & | input2_shape, | ||
const T * | input2_data, | ||
const core::OMRuntimeShape & | output_shape, | ||
T * | output_data | ||
) |
Definition at line 58 of file PALAddCommon.h.
References onert_micro::Ok, and output_shape.
OMStatus onert_micro::execute::pal::BroadcastArithmeticOp4DSlow | ( | const core::BinaryArithmeticBroadcastParams & | params, |
const core::OMRuntimeShape & | input1_shape, | ||
const T * | input1_data, | ||
const core::OMRuntimeShape & | input2_shape, | ||
const T * | input2_data, | ||
const core::OMRuntimeShape & | output_shape, | ||
T * | output_data | ||
) |
Definition at line 93 of file PALArithmeticOpCommon.h.
References desc1, desc2, onert_micro::core::OMRuntimeShape::dims(), onert_micro::core::OMRuntimeShape::extendedShape(), getActivationParams(), NdArrayDescsForElementwiseBroadcast(), onert_micro::Ok, output_shape, and subscriptToIndex().
void onert_micro::execute::pal::BroadcastBinaryFunction6DSlow | ( | const core::ArithmeticQuantParams & | params, |
const core::OMRuntimeShape & | input1_shape, | ||
const T * | input1_data, | ||
const core::OMRuntimeShape & | input2_shape, | ||
const T * | input2_data, | ||
const core::OMRuntimeShape & | output_shape, | ||
T * | output_data, | ||
T(*)(T, T, const core::ArithmeticQuantParams &) | binary_func | ||
) |
Definition at line 224 of file PALArithmeticOpCommon.h.
References BroadcastRecursiveDimensions().
Referenced by BroadcastAdd4DSlow(), and BroadcastSub4DSlow().
|
inline |
Definition at line 75 of file PALBinaryOpCommon.h.
References desc1, desc2, onert_micro::core::OMRuntimeShape::dims(), onert_micro::core::OMRuntimeShape::extendedShape(), NdArrayDescsForElementwiseBroadcast(), onert_micro::Ok, output_shape, and subscriptToIndex().
|
inline |
Definition at line 149 of file PALComparisons.h.
References subscriptToIndex().
|
inline |
Definition at line 75 of file PALComparisons.h.
References onert_micro::core::ComparisonParams::input1_multiplier, onert_micro::core::ComparisonParams::input1_offset, onert_micro::core::ComparisonParams::input1_shift, onert_micro::core::ComparisonParams::input2_multiplier, onert_micro::core::ComparisonParams::input2_offset, onert_micro::core::ComparisonParams::input2_shift, onert_micro::core::ComparisonParams::left_shift, multiplyByQuantizedMultiplierSmallerThanOneExp(), and subscriptToIndex().
OMStatus onert_micro::execute::pal::BroadcastDiv4DSlow | ( | const core::BinaryArithmeticBroadcastParams & | params, |
const core::OMRuntimeShape & | input1_shape, | ||
const T * | input1_data, | ||
const core::OMRuntimeShape & | input2_shape, | ||
const T * | input2_data, | ||
const core::OMRuntimeShape & | output_shape, | ||
T * | output_data | ||
) |
Definition at line 39 of file PALDivCommon.h.
References onert_micro::Ok, and output_shape.
|
inline |
Definition at line 40 of file PALFloorDivCommon.h.
References output_shape.
|
inline |
Definition at line 40 of file PALFloorModCommon.h.
References output_shape.
void onert_micro::execute::pal::BroadcastInput1 | ( | int | size, |
const core::ArithmeticQuantParams & | params, | ||
const T * | input1_data, | ||
const T * | input2_data, | ||
T * | output_data, | ||
T(*)(T, T, const core::ArithmeticQuantParams &) | binary_func | ||
) |
Definition at line 145 of file PALArithmeticOpCommon.h.
References size.
void onert_micro::execute::pal::BroadcastInput2 | ( | int | size, |
const core::ArithmeticQuantParams & | params, | ||
const T * | input1_data, | ||
const T * | input2_data, | ||
T * | output_data, | ||
T(*)(T, T, const core::ArithmeticQuantParams &) | binary_func | ||
) |
Definition at line 156 of file PALArithmeticOpCommon.h.
References size.
|
inline |
Definition at line 38 of file PALMaximumCommon.h.
References output_shape.
|
inline |
Definition at line 40 of file PALMinimumCommon.h.
References output_shape.
OMStatus onert_micro::execute::pal::BroadcastMul4DSlow | ( | const core::BinaryArithmeticBroadcastParams & | params, |
const core::OMRuntimeShape & | input1_shape, | ||
const T * | input1_data, | ||
const core::OMRuntimeShape & | input2_shape, | ||
const T * | input2_data, | ||
const core::OMRuntimeShape & | output_shape, | ||
T * | output_data | ||
) |
Definition at line 44 of file PALMulCommon.h.
References onert_micro::Ok, and output_shape.
OMStatus onert_micro::execute::pal::BroadcastMul6DSlow | ( | const core::ArithmeticQuantParams & | params, |
const core::OMRuntimeShape & | input1_shape, | ||
const T * | input1_data, | ||
const core::OMRuntimeShape & | input2_shape, | ||
const T * | input2_data, | ||
const core::OMRuntimeShape & | output_shape, | ||
T * | output_data | ||
) |
Definition at line 55 of file PALMulCommon.h.
References desc1, desc2, onert_micro::core::OMRuntimeShape::dimsData(), onert_micro::core::OMRuntimeShape::extendedShape(), onert_micro::core::ArithmeticQuantParams::input1_offset, onert_micro::core::ArithmeticQuantParams::input2_offset, multiplyByQuantizedMultiplier(), NdArrayDescsForElementwiseBroadcast(), onert_micro::Ok, onert_micro::core::ArithmeticQuantParams::output_multiplier, onert_micro::core::ArithmeticQuantParams::output_offset, output_shape, onert_micro::core::ArithmeticQuantParams::output_shift, onert_micro::core::ArithmeticQuantParams::quantized_activation_max, and onert_micro::core::ArithmeticQuantParams::quantized_activation_min.
void onert_micro::execute::pal::BroadcastRecursiveDimensions | ( | const core::ArithmeticQuantParams & | params, |
int | dimension, | ||
size_t * | input1_offset_p, | ||
size_t * | input2_offset_p, | ||
size_t * | output_offset, | ||
size_t * | compressed_input1_stride, | ||
size_t * | compressed_input2_stride, | ||
size_t * | compressed_output_shape, | ||
const T * | input1_data, | ||
const T * | input2_data, | ||
T * | output_data, | ||
T(*)(T, T, const core::ArithmeticQuantParams &) | binary_func | ||
) |
Definition at line 167 of file PALArithmeticOpCommon.h.
References BroadcastRecursiveDimensions().
Referenced by BroadcastBinaryFunction6DSlow(), and BroadcastRecursiveDimensions().
OMStatus onert_micro::execute::pal::BroadcastSquaredDifference4DSlow | ( | const core::BinaryArithmeticBroadcastParams & | params, |
const core::OMRuntimeShape & | input1_shape, | ||
const T * | input1_data, | ||
const core::OMRuntimeShape & | input2_shape, | ||
const T * | input2_data, | ||
const core::OMRuntimeShape & | output_shape, | ||
T * | output_data | ||
) |
Definition at line 38 of file PALSquaredDifferenceCommon.h.
References onert_micro::Ok, and output_shape.
OMStatus onert_micro::execute::pal::BroadcastSub4DSlow | ( | const core::ArithmeticQuantParams & | params, |
const core::OMRuntimeShape & | input1_shape, | ||
const int8_t * | input1_data, | ||
const core::OMRuntimeShape & | input2_shape, | ||
const int8_t * | input2_data, | ||
const core::OMRuntimeShape & | output_shape, | ||
int8_t * | output_data | ||
) |
Definition at line 67 of file PALSubCommon.h.
References BroadcastBinaryFunction6DSlow(), onert_micro::Ok, output_shape, and SubFunc().
OMStatus onert_micro::execute::pal::BroadcastSub4DSlow | ( | const core::BinaryArithmeticBroadcastParams & | params, |
const core::OMRuntimeShape & | input1_shape, | ||
const T * | input1_data, | ||
const core::OMRuntimeShape & | input2_shape, | ||
const T * | input2_data, | ||
const core::OMRuntimeShape & | output_shape, | ||
T * | output_data | ||
) |
Definition at line 57 of file PALSubCommon.h.
References onert_micro::Ok, and output_shape.
|
inline |
Definition at line 34 of file PALCast.h.
References onert_micro::core::OMRuntimeShape::flatSize(), onert_micro::Ok, output_shape, and onert_micro::UnknownError.
|
inline |
Definition at line 34 of file PALCeil.h.
References onert_micro::core::OMRuntimeShape::flatSize(), onert_micro::Ok, output_shape, and onert_micro::UnknownError.
|
inline |
Definition at line 65 of file PALComparisons.h.
|
inline |
Definition at line 122 of file PALComparisons.h.
References onert_micro::core::ComparisonParams::input1_multiplier, onert_micro::core::ComparisonParams::input1_offset, onert_micro::core::ComparisonParams::input1_shift, onert_micro::core::ComparisonParams::input2_multiplier, onert_micro::core::ComparisonParams::input2_offset, onert_micro::core::ComparisonParams::input2_shift, onert_micro::core::ComparisonParams::left_shift, and multiplyByQuantizedMultiplierSmallerThanOneExp().
OMStatus onert_micro::execute::pal::Concatenation | ( | core::ConcatenationParams & | params, |
std::vector< uint32_t > & | input_shapes, | ||
std::vector< const Scalar * > & | input_data, | ||
const core::OMRuntimeShape & | output_shape, | ||
Scalar * | output_data | ||
) |
Definition at line 30 of file PALConcatenation.h.
References onert_micro::core::ConcatenationParams::axis, luci_interpreter::RuntimeShape::dimensionsCount(), luci_interpreter::RuntimeShape::dims(), onert_micro::core::ConcatenationParams::num_inputs, onert_micro::Ok, and output_shape.
OMStatus onert_micro::execute::pal::ConvFloat | ( | const core::FloatConv2D * | params, |
const core::OMRuntimeShape & | input_shape, | ||
const float * | input_data, | ||
const core::OMRuntimeShape & | filter_shape, | ||
const float * | filter_data, | ||
const float * | bias_data, | ||
const core::OMRuntimeShape & | output_shape, | ||
float * | output_data | ||
) |
Definition at line 31 of file PALConv2DCommon.h.
References onert_micro::core::FloatConv2D::activation_max, onert_micro::core::FloatConv2D::activation_min, onert_micro::core::FloatConv2D::dilation_height_factor, onert_micro::core::FloatConv2D::dilation_width_factor, luci_interpreter::RuntimeShape::dims(), onert_micro::core::OMRuntimeShape::dims(), onert_micro::Ok, output_shape, onert_micro::core::FloatConv2D::pad_h, onert_micro::core::FloatConv2D::pad_w, onert_micro::core::FloatConv2D::stride_h, and onert_micro::core::FloatConv2D::stride_w.
OMStatus onert_micro::execute::pal::ConvPerChannel | ( | const core::ConvQuant & | params, |
const core::OMRuntimeShape & | input_shape, | ||
const int8_t * | input_data, | ||
const core::OMRuntimeShape & | filter_shape, | ||
const int8_t * | filter_data, | ||
const int32_t * | bias_data, | ||
const core::OMRuntimeShape & | output_shape, | ||
int8_t * | output_data | ||
) |
Definition at line 36 of file PALConv2D.h.
References onert_micro::CmsisNNError, onert_micro::core::ConvQuant::dilation_height_factor, onert_micro::core::ConvQuant::dilation_width_factor, luci_interpreter::RuntimeShape::dims(), onert_micro::core::OMRuntimeShape::dims(), onert_micro::core::ConvQuant::input_offset, onert_micro::Ok, onert_micro::core::ConvQuant::output_offset, output_shape, onert_micro::core::ConvQuant::pad_h, onert_micro::core::ConvQuant::pad_w, onert_micro::core::ConvQuant::per_channel_output_multiplier, onert_micro::core::ConvQuant::per_channel_output_shift, onert_micro::core::ConvQuant::quantized_activation_max, onert_micro::core::ConvQuant::quantized_activation_min, onert_micro::core::ConvQuant::stride_height, and onert_micro::core::ConvQuant::stride_width.
|
inline |
Definition at line 53 of file ProcessBroadcastShapes.h.
References onert_micro::core::OMRuntimeShape::dims(), onert_micro::execute::pal::NdArrayDesc< N >::extents, and onert_micro::execute::pal::NdArrayDesc< N >::strides.
Referenced by TransposeImpl().
|
inline |
Definition at line 34 of file PALCosCommon.h.
References onert_micro::core::OMRuntimeShape::flatSize(), onert_micro::Ok, output_shape, and onert_micro::UnknownError.
|
inline |
Definition at line 34 of file PALDepthwiseConv2DCommon.h.
|
inline |
Definition at line 41 of file PALDepthwiseConv2DCommon.h.
References onert_micro::core::FloatConv2D::activation_max, onert_micro::core::FloatConv2D::activation_min, activationFunctionWithMinMax(), onert_micro::core::FloatConv2D::depth_multiplier, onert_micro::core::FloatConv2D::dilation_height_factor, onert_micro::core::FloatConv2D::dilation_width_factor, luci_interpreter::RuntimeShape::dims(), onert_micro::core::OMRuntimeShape::dims(), luci_interpreter::RuntimeShape::dimsData(), onert_micro::core::OMRuntimeShape::dimsData(), m, offset(), onert_micro::Ok, output_shape, onert_micro::core::FloatConv2D::pad_h, onert_micro::core::FloatConv2D::pad_w, onert_micro::core::FloatConv2D::stride_h, and onert_micro::core::FloatConv2D::stride_w.
OMStatus onert_micro::execute::pal::DepthwiseConvPerChannel | ( | const core::ConvQuant & | params, |
const core::OMRuntimeShape & | input_shape, | ||
const int8_t * | input_data, | ||
const core::OMRuntimeShape & | filter_shape, | ||
const int8_t * | filter_data, | ||
const int32_t * | bias_data, | ||
const core::OMRuntimeShape & | output_shape, | ||
int8_t * | output_data | ||
) |
Definition at line 36 of file PALDepthwiseConv2D.h.
References onert_micro::CmsisNNError, onert_micro::core::ConvQuant::depth_multiplier, onert_micro::core::ConvQuant::dilation_height_factor, onert_micro::core::ConvQuant::dilation_width_factor, luci_interpreter::RuntimeShape::dims(), onert_micro::core::OMRuntimeShape::dims(), onert_micro::core::ConvQuant::input_offset, onert_micro::Ok, onert_micro::core::ConvQuant::output_offset, output_shape, onert_micro::core::ConvQuant::pad_h, onert_micro::core::ConvQuant::pad_w, onert_micro::core::ConvQuant::per_channel_output_multiplier, onert_micro::core::ConvQuant::per_channel_output_shift, onert_micro::core::ConvQuant::quantized_activation_max, onert_micro::core::ConvQuant::quantized_activation_min, onert_micro::core::ConvQuant::stride_height, and onert_micro::core::ConvQuant::stride_width.
OMStatus onert_micro::execute::pal::Dequantize | ( | const core::QuantizationParams | op_params, |
const uint32_t | flat_size, | ||
const InputT * | input_data, | ||
OutputT * | output_data | ||
) |
Definition at line 35 of file PALDequantize.h.
References onert_micro::Ok, onert_micro::core::QuantizationParams::scale, and onert_micro::core::QuantizationParams::zero_point.
OMStatus onert_micro::execute::pal::Div | ( | const core::BinaryArithmeticBroadcastParams & | params, |
const int | flat_size, | ||
const T * | input1_data, | ||
const T * | input2_data, | ||
T * | output_data | ||
) |
Definition at line 31 of file PALDivCommon.h.
References onert_micro::Ok.
void onert_micro::execute::pal::ElementWise | ( | const uint32_t | size, |
const core::ArithmeticQuantParams & | params, | ||
const T * | input1_data, | ||
const T * | input2_data, | ||
T * | output_data, | ||
T(*)(T, T, const core::ArithmeticQuantParams &) | binary_func | ||
) |
Definition at line 69 of file PALArithmeticOpCommon.h.
References size.
Referenced by Sub().
|
inline |
Definition at line 32 of file PALElu.h.
References onert_micro::Ok.
|
inline |
Definition at line 59 of file PALComparisons.h.
|
inline |
Definition at line 34 of file PALExpCommon.h.
References onert_micro::core::OMRuntimeShape::flatSize(), onert_micro::Ok, output_shape, and onert_micro::UnknownError.
|
inline |
Definition at line 32 of file PALFill.h.
References luci_interpreter::RuntimeShape::flatSize(), onert_micro::Ok, output_shape, and onert_micro::UnknownError.
|
inline |
Definition at line 210 of file PALUtils.h.
Referenced by FullyConnected(), FullyConnected(), FullyConnected(), and FullyConnected< int8_t >().
|
inline |
Definition at line 33 of file PALFloorCommon.h.
References luci_interpreter::RuntimeShape::dimensionsCount(), onert_micro::core::OMRuntimeShape::dimensionsCount(), luci_interpreter::RuntimeShape::dims(), onert_micro::core::OMRuntimeShape::dims(), onert_micro::core::OMRuntimeShape::flatSize(), onert_micro::Ok, output_shape, and onert_micro::UnknownError.
|
inline |
Definition at line 32 of file PALFloorDivCommon.h.
|
inline |
Definition at line 32 of file PALFloorModCommon.h.
|
inline |
Definition at line 80 of file PALFullyConnectedCommon.h.
References luci_interpreter::RuntimeShape::dimensionsCount(), onert_micro::core::OMRuntimeShape::dimensionsCount(), luci_interpreter::RuntimeShape::dims(), onert_micro::core::OMRuntimeShape::dims(), luci_interpreter::RuntimeShape::dimsData(), flatSizeSkipDim(), onert_micro::core::FullyConnectedParams::float_activation_max, onert_micro::core::FullyConnectedParams::float_activation_min, onert_micro::core::FullyConnectedParams::is_channel_wise_quant, onert_micro::Ok, output_shape, and onert_micro::core::FullyConnectedParams::weights_scales.
OMStatus onert_micro::execute::pal::FullyConnected | ( | const core::FullyConnectedParams & | params, |
const InputType * | input_data, | ||
const core::OMRuntimeShape & | filter_shape, | ||
const WeightType * | filter_data, | ||
const BiasType * | bias_data, | ||
const core::OMRuntimeShape & | output_shape, | ||
OutputType * | output_data | ||
) |
Definition at line 34 of file PALFullyConnectedCommon.h.
References luci_interpreter::RuntimeShape::dimensionsCount(), onert_micro::core::OMRuntimeShape::dimensionsCount(), luci_interpreter::RuntimeShape::dims(), onert_micro::core::OMRuntimeShape::dims(), luci_interpreter::RuntimeShape::dimsData(), flatSizeSkipDim(), onert_micro::core::FullyConnectedParams::input_offset, multiplyByQuantizedMultiplier(), onert_micro::Ok, onert_micro::core::FullyConnectedParams::output_multiplier, onert_micro::core::FullyConnectedParams::output_offset, output_shape, onert_micro::core::FullyConnectedParams::output_shift, onert_micro::core::FullyConnectedParams::quantized_activation_max, onert_micro::core::FullyConnectedParams::quantized_activation_min, and onert_micro::core::FullyConnectedParams::weights_offset.
OMStatus onert_micro::execute::pal::FullyConnected | ( | const core::FullyConnectedParams & | params, |
const int16_t * | input_data, | ||
const core::OMRuntimeShape & | filter_shape, | ||
const int8_t * | filter_data, | ||
const int64_t * | bias_data, | ||
const core::OMRuntimeShape & | output_shape, | ||
int16_t * | output_data | ||
) |
Definition at line 98 of file PALFullyConnected.h.
References onert_micro::CmsisNNError, luci_interpreter::RuntimeShape::dimensionsCount(), onert_micro::core::OMRuntimeShape::dimensionsCount(), luci_interpreter::RuntimeShape::dims(), onert_micro::core::OMRuntimeShape::dims(), luci_interpreter::RuntimeShape::dimsData(), flatSizeSkipDim(), onert_micro::core::FullyConnectedParams::input_offset, onert_micro::Ok, onert_micro::core::FullyConnectedParams::output_multiplier, onert_micro::core::FullyConnectedParams::output_offset, output_shape, onert_micro::core::FullyConnectedParams::output_shift, onert_micro::core::FullyConnectedParams::quantized_activation_max, onert_micro::core::FullyConnectedParams::quantized_activation_min, and onert_micro::core::FullyConnectedParams::weights_offset.
OMStatus onert_micro::execute::pal::FullyConnected< int8_t > | ( | const core::FullyConnectedParams & | params, |
const int8_t * | input_data, | ||
const core::OMRuntimeShape & | filter_shape, | ||
const int8_t * | filter_data, | ||
const int32_t * | bias_data, | ||
const core::OMRuntimeShape & | output_shape, | ||
int8_t * | output_data | ||
) |
Definition at line 32 of file PALFullyConnected.h.
References onert_micro::CmsisNNError, luci_interpreter::RuntimeShape::dimensionsCount(), onert_micro::core::OMRuntimeShape::dimensionsCount(), luci_interpreter::RuntimeShape::dims(), onert_micro::core::OMRuntimeShape::dims(), luci_interpreter::RuntimeShape::dimsData(), flatSizeSkipDim(), onert_micro::core::FullyConnectedParams::input_offset, onert_micro::Ok, onert_micro::core::FullyConnectedParams::output_multiplier, onert_micro::core::FullyConnectedParams::output_offset, output_shape, onert_micro::core::FullyConnectedParams::output_shift, onert_micro::core::FullyConnectedParams::quantized_activation_max, onert_micro::core::FullyConnectedParams::quantized_activation_min, and onert_micro::core::FullyConnectedParams::weights_offset.
|
inline |
Definition at line 35 of file PALGatherND.h.
References onert_micro::core::OMRuntimeShape::dimensionsCount(), onert_micro::core::OMRuntimeShape::dims(), onert_micro::core::OMRuntimeShape::flatSize(), MAX_INDICES_ND, offset(), onert_micro::Ok, and onert_micro::UnknownError.
|
inline |
Definition at line 126 of file PALUtils.h.
|
inline |
Definition at line 120 of file PALUtils.h.
Referenced by ArithmeticOp(), ArithmeticOpScalar(), and BroadcastArithmeticOp4DSlow().
|
inline |
Definition at line 132 of file PALUtils.h.
|
inline |
Definition at line 30 of file PALUtils.h.
References onert_micro::core::ALL, onert_micro::core::LOWER_1_2_PART, and onert_micro::core::UP_1_2_PART.
Referenced by onert_micro::train::pal::Conv2DWeightGrad(), and onert_micro::train::pal::FullyConnectedWeightGrad().
|
inline |
Definition at line 61 of file PALComparisons.h.
|
inline |
Definition at line 60 of file PALComparisons.h.
OMStatus onert_micro::execute::pal::GRU | ( | const float * | input_data, |
const float * | weight_input_data, | ||
const float * | weight_hidden_data, | ||
const float * | bias_input_data, | ||
const float * | bias_hidden_data, | ||
const float * | hidden_state_data, | ||
float * | output_data, | ||
float * | output_input_data, | ||
float * | output_hidden_data, | ||
const core::OMRuntimeShape & | input_shape, | ||
const core::OMRuntimeShape & | output_shape, | ||
const core::OMRuntimeShape & | weight_input_shape, | ||
const core::OMRuntimeShape & | weight_hidden_shape, | ||
const size_t | intermediate_buffer_size, | ||
float * | intermediate_buffer | ||
) |
Definition at line 183 of file PALGRUCommon.h.
References onert_micro::core::OMRuntimeShape::dims(), luci_interpreter::RuntimeShape::flatSize(), onert_micro::Ok, output_shape, and onert_micro::core::OMRuntimeShape::setDim().
|
inline |
Definition at line 33 of file PALL2Normalize.h.
References onert_micro::core::L2NormalizationParams::epsilon, onert_micro::core::L2NormalizationParams::num_rows, onert_micro::Ok, and onert_micro::core::L2NormalizationParams::row_size.
OMStatus onert_micro::execute::pal::L2Pool | ( | const core::Pool2DParams & | params, |
const core::OMRuntimeShape & | input_shape, | ||
const float * | input_data, | ||
const core::OMRuntimeShape & | output_shape, | ||
float * | output_data | ||
) |
Definition at line 34 of file PALL2Pool2DCommon.h.
References onert_micro::core::Pool2DParams::activation_max, onert_micro::core::Pool2DParams::activation_min, activationFunctionWithMinMax(), luci_interpreter::RuntimeShape::dims(), onert_micro::core::OMRuntimeShape::dims(), luci_interpreter::RuntimeShape::dimsData(), onert_micro::core::OMRuntimeShape::dimsData(), onert_micro::FailedCheckCondition, onert_micro::core::Pool2DParams::filter_h, onert_micro::core::Pool2DParams::filter_w, offset(), onert_micro::Ok, output_shape, onert_micro::core::Pool2DParams::pad_h, onert_micro::core::Pool2DParams::pad_w, onert_micro::core::Pool2DParams::stride_h, and onert_micro::core::Pool2DParams::stride_w.
|
inline |
Definition at line 58 of file PALComparisons.h.
|
inline |
Definition at line 57 of file PALComparisons.h.
|
inline |
Definition at line 34 of file PALLogCommon.h.
References onert_micro::core::OMRuntimeShape::flatSize(), onert_micro::Ok, output_shape, and onert_micro::UnknownError.
|
inline |
Definition at line 32 of file PALLogistic.h.
References onert_micro::Ok.
|
inline |
Definition at line 66 of file PALLogistic.h.
References onert_micro::Ok.
|
inline |
Definition at line 32 of file PALLogSoftmax.h.
References onert_micro::core::LogSoftmaxParams::num_rows, onert_micro::Ok, and onert_micro::core::LogSoftmaxParams::row_size.
|
inline |
Definition at line 200 of file PALUtils.h.
References onert_micro::core::OMRuntimeShape::dims().
|
inline |
Definition at line 32 of file PALMaximumCommon.h.
OMStatus onert_micro::execute::pal::MaxPool | ( | const core::Pool2DParams & | params, |
const core::OMRuntimeShape & | input_shape, | ||
const float * | input_data, | ||
const core::OMRuntimeShape & | output_shape, | ||
float * | output_data | ||
) |
Definition at line 32 of file PALMaxPool2DCommon.h.
References onert_micro::core::Pool2DParams::activation_max, onert_micro::core::Pool2DParams::activation_min, luci_interpreter::RuntimeShape::dims(), onert_micro::core::OMRuntimeShape::dims(), onert_micro::core::Pool2DParams::filter_h, onert_micro::core::Pool2DParams::filter_w, onert_micro::Ok, output_shape, onert_micro::core::Pool2DParams::pad_h, onert_micro::core::Pool2DParams::pad_w, onert_micro::core::Pool2DParams::stride_h, and onert_micro::core::Pool2DParams::stride_w.
OMStatus onert_micro::execute::pal::MaxPool | ( | const core::Pool2DParams & | params, |
const core::OMRuntimeShape & | input_shape, | ||
const int8_t * | input_data, | ||
const core::OMRuntimeShape & | output_shape, | ||
int8_t * | output_data | ||
) |
Definition at line 32 of file PALMaxPool2D.h.
References onert_micro::CmsisNNError, luci_interpreter::RuntimeShape::dims(), onert_micro::core::OMRuntimeShape::dims(), onert_micro::core::Pool2DParams::filter_h, onert_micro::core::Pool2DParams::filter_w, onert_micro::Ok, output_shape, onert_micro::core::Pool2DParams::pad_h, onert_micro::core::Pool2DParams::pad_w, onert_micro::core::Pool2DParams::quantized_activation_max, onert_micro::core::Pool2DParams::quantized_activation_min, onert_micro::core::Pool2DParams::stride_h, and onert_micro::core::Pool2DParams::stride_w.
|
inline |
Definition at line 32 of file PALMinimumCommon.h.
OMStatus onert_micro::execute::pal::Mul | ( | const core::ArithmeticQuantParams & | params, |
const uint32_t | flat_size, | ||
const int8_t * | input1_data, | ||
const int8_t * | input2_data, | ||
int8_t * | output_data | ||
) |
Definition at line 33 of file PALMul.h.
References onert_micro::core::ArithmeticQuantParams::input1_offset, onert_micro::core::ArithmeticQuantParams::input2_offset, onert_micro::Ok, onert_micro::core::ArithmeticQuantParams::output_multiplier, onert_micro::core::ArithmeticQuantParams::output_offset, onert_micro::core::ArithmeticQuantParams::output_shift, onert_micro::core::ArithmeticQuantParams::quantized_activation_max, onert_micro::core::ArithmeticQuantParams::quantized_activation_min, and onert_micro::UnknownError.
OMStatus onert_micro::execute::pal::Mul | ( | const core::ArithmeticQuantParams & | params, |
uint32_t | size, | ||
const InputType * | input1_data, | ||
const InputType * | input2_data, | ||
OutputType * | output_data | ||
) |
Definition at line 31 of file PALMul.h.
References onert_micro::core::ArithmeticQuantParams::input1_offset, onert_micro::core::ArithmeticQuantParams::input2_offset, multiplyByQuantizedMultiplier(), onert_micro::Ok, onert_micro::core::ArithmeticQuantParams::output_multiplier, onert_micro::core::ArithmeticQuantParams::output_offset, onert_micro::core::ArithmeticQuantParams::output_shift, onert_micro::core::ArithmeticQuantParams::quantized_activation_max, onert_micro::core::ArithmeticQuantParams::quantized_activation_min, and size.
OMStatus onert_micro::execute::pal::Mul | ( | const core::BinaryArithmeticBroadcastParams & | params, |
const int | flat_size, | ||
const T * | input1_data, | ||
const T * | input2_data, | ||
T * | output_data | ||
) |
Definition at line 36 of file PALMulCommon.h.
References onert_micro::Ok.
|
inline |
Definition at line 104 of file PALUtils.h.
References roundingDivideByPOT(), and saturatingRoundingDoublingHighMul().
Referenced by BroadcastMul6DSlow(), FullyConnected(), and Mul().
|
inline |
Definition at line 112 of file PALUtils.h.
References roundingDivideByPOT(), and saturatingRoundingDoublingHighMul().
Referenced by AddFunc(), BroadcastComparison4DSlowWithScaling(), ComparisonWithScaling(), and SubFunc().
|
inline |
Definition at line 94 of file ProcessBroadcastShapes.h.
References onert_micro::core::OMRuntimeShape::extendedShape(), onert_micro::execute::pal::NdArrayDesc< N >::extents, and onert_micro::execute::pal::NdArrayDesc< N >::strides.
Referenced by BroadcastArithmeticOp4DSlow(), BroadcastBinaryOp4DSlow(), and BroadcastMul6DSlow().
|
inline |
Definition at line 87 of file ProcessBroadcastShapes.h.
std::enable_if< DIM==N-1, void >::type onert_micro::execute::pal::NDOpsHelperImpl | ( | const NdArrayDesc< N > & | output, |
const Calc & | calc, | ||
int | indexes[N] | ||
) |
Definition at line 65 of file ProcessBroadcastShapes.h.
std::enable_if< DIM!=N-1, void >::type onert_micro::execute::pal::NDOpsHelperImpl | ( | const NdArrayDesc< N > & | output, |
const Calc & | calc, | ||
int | indexes[N] | ||
) |
Definition at line 75 of file ProcessBroadcastShapes.h.
|
inline |
Definition at line 30 of file PALNegCommon.h.
References onert_micro::core::OMRuntimeShape::flatSize(), onert_micro::Ok, output_shape, and onert_micro::UnknownError.
|
inline |
Definition at line 175 of file PALUtils.h.
|
inline |
Definition at line 62 of file PALComparisons.h.
|
inline |
Definition at line 220 of file PALUtils.h.
Referenced by BatchToSpaceND(), DepthwiseConv2D< float >(), GatherND(), L2Pool(), reducedOutputOffset(), SpaceToBatchND(), SpaceToDepth(), and TransposeConv< float >().
|
inline |
Definition at line 225 of file PALUtils.h.
OMStatus onert_micro::execute::pal::Pad | ( | const core::PadParams & | op_params, |
const core::OMRuntimeShape & | input_shape, | ||
const float * | input_data, | ||
const float | pad_value, | ||
const core::OMRuntimeShape & | output_shape, | ||
float * | output_data | ||
) |
Definition at line 35 of file PALPad.h.
References onert_micro::core::OMRuntimeShape::extendedShape(), onert_micro::core::PadParams::left_padding, onert_micro::core::PadParams::left_padding_count, onert_micro::Ok, output_shape, onert_micro::core::PadParams::right_padding, and onert_micro::core::PadParams::right_padding_count.
|
inline |
Definition at line 155 of file ProcessBroadcastShapes.h.
References onert_micro::core::BinaryArithmeticBroadcastParams::broadcast_category, onert_micro::core::OMRuntimeShape::dimensionsCount(), onert_micro::core::OMRuntimeShape::extendedShape(), onert_micro::core::OMRuntimeShape::flatSize(), onert_micro::core::kFirstInputBroadcastsFast, onert_micro::core::kGenericBroadcast, onert_micro::core::kNonBroadcast, onert_micro::core::kScalarFirstBroadcast, onert_micro::core::kScalarSecondBroadcast, and onert_micro::core::kSecondInputBroadcastsFast.
OMStatus onert_micro::execute::pal::Quantize | ( | const core::QuantizationParams | op_params, |
const uint32_t | flat_size, | ||
const InputT * | input_data, | ||
OutputT * | output_data | ||
) |
Definition at line 35 of file PALQuantize.h.
References onert_micro::Ok, onert_micro::core::QuantizationParams::scale, and onert_micro::core::QuantizationParams::zero_point.
bool onert_micro::execute::pal::ReduceDimensionsForBroadcast | ( | const core::OMRuntimeShape & | input1_shape, |
const core::OMRuntimeShape & | input2_shape, | ||
size_t * | compressed_input1_stride, | ||
size_t * | compressed_input2_stride, | ||
size_t * | compressed_output_shape | ||
) |
Definition at line 242 of file PALUtils.h.
References onert_micro::core::OMRuntimeShape::dimensionsCount(), and onert_micro::core::OMRuntimeShape::dimsData().
|
inline |
Definition at line 143 of file PALUtils.h.
References offset().
OMStatus onert_micro::execute::pal::ReLUCommon | ( | const int | flat_size, |
const Type * | input_data, | ||
Type * | output_data, | ||
const float | alpha, | ||
const bool | is_relu_6 | ||
) |
Definition at line 33 of file PALReluCommon.h.
References onert_micro::Ok.
Referenced by onert_micro::execute::execute_relu_common().
OMStatus onert_micro::execute::pal::ReLUCommon< int8_t > | ( | const int | flat_size, |
const int8_t * | input_data, | ||
int8_t * | output_data, | ||
const float | alpha, | ||
const bool | is_relu_6 | ||
) |
Definition at line 33 of file PALRelu.h.
References onert_micro::Ok.
|
inline |
|
inline |
Definition at line 38 of file PALRoundCommon.h.
References onert_micro::core::OMRuntimeShape::flatSize(), onert_micro::Ok, output_shape, and onert_micro::UnknownError.
|
inline |
Definition at line 92 of file PALUtils.h.
Referenced by multiplyByQuantizedMultiplier(), and multiplyByQuantizedMultiplierSmallerThanOneExp().
|
inline |
Definition at line 34 of file PALRsqrtCommon.h.
References output_shape.
|
inline |
Definition at line 79 of file PALUtils.h.
Referenced by multiplyByQuantizedMultiplier(), and multiplyByQuantizedMultiplierSmallerThanOneExp().
|
inline |
Definition at line 35 of file PALSinCommon.h.
References onert_micro::core::OMRuntimeShape::flatSize(), onert_micro::Ok, output_shape, and onert_micro::UnknownError.
|
inline |
Definition at line 31 of file PALSISOOperation.h.
References onert_micro::core::OMRuntimeShape::flatSize(), onert_micro::Ok, output_shape, and onert_micro::UnknownError.
OMStatus onert_micro::execute::pal::Slice | ( | const core::SliceParams & | op_params, |
const core::OMRuntimeShape & | input_shape, | ||
const T * | input_data, | ||
T * | output_data | ||
) |
Definition at line 29 of file PALSlice.h.
References onert_micro::core::SliceParams::begin, begin_count, onert_micro::core::SliceParams::begin_count, onert_micro::core::OMRuntimeShape::dims(), onert_micro::core::OMRuntimeShape::extendedShape(), onert_micro::Ok, onert_micro::core::SliceParams::size, size_count, and onert_micro::core::SliceParams::size_count.
OMStatus onert_micro::execute::pal::Softmax | ( | const core::SoftmaxParams & | params, |
const T * | input_data, | ||
U * | output_data | ||
) |
Definition at line 33 of file PALSoftmaxCommon.h.
References onert_micro::core::SoftmaxParams::beta, onert_micro::core::SoftmaxParams::input_scale, onert_micro::core::SoftmaxParams::input_zp, onert_micro::core::SoftmaxParams::num_rows, onert_micro::Ok, onert_micro::core::SoftmaxParams::output_scale, onert_micro::core::SoftmaxParams::output_zp, and onert_micro::core::SoftmaxParams::row_size.
OMStatus onert_micro::execute::pal::Softmax< float, float > | ( | const core::SoftmaxParams & | params, |
const float * | input_data, | ||
float * | output_data | ||
) |
Definition at line 86 of file PALSoftmaxCommon.h.
References onert_micro::core::SoftmaxParams::beta, onert_micro::core::SoftmaxParams::num_rows, onert_micro::Ok, onert_micro::core::SoftmaxParams::row_size, and onert_micro::UnknownError.
OMStatus onert_micro::execute::pal::Softmax< int8_t, int8_t > | ( | const core::SoftmaxParams & | params, |
const int8_t * | input_data, | ||
int8_t * | output_data | ||
) |
Definition at line 33 of file PALSoftmax.h.
References onert_micro::core::SoftmaxParams::diff_min, onert_micro::core::SoftmaxParams::input_left_shift, onert_micro::core::SoftmaxParams::input_multiplier, onert_micro::core::SoftmaxParams::num_rows, onert_micro::Ok, and onert_micro::core::SoftmaxParams::row_size.
|
inline |
Definition at line 51 of file PALSpaceToBatchNDCommon.h.
References onert_micro::core::OMRuntimeShape::dimensionsCount(), luci_interpreter::RuntimeShape::dims(), onert_micro::core::OMRuntimeShape::dims(), luci_interpreter::RuntimeShape::dimsData(), onert_micro::core::OMRuntimeShape::dimsData(), offset(), onert_micro::Ok, and output_shape.
|
inline |
Definition at line 32 of file PALSpaceToDepthCommon.h.
References onert_micro::core::OMRuntimeShape::dims(), luci_interpreter::RuntimeShape::dimsData(), onert_micro::core::OMRuntimeShape::dimsData(), onert_micro::core::OMRuntimeShape::extendedShape(), onert_micro::FailedCheckCondition, offset(), onert_micro::Ok, and output_shape.
OMStatus onert_micro::execute::pal::Split | ( | const core::SplitParams & | params, |
const core::OMRuntimeShape & | input_shape, | ||
const T * | input_data, | ||
const core::OMRuntimeShape & | output_shape, | ||
int32_t | axis_value | ||
) |
Definition at line 36 of file PALSplit.h.
References luci_interpreter::RuntimeShape::dimensionsCount(), onert_micro::core::OMRuntimeShape::dimensionsCount(), luci_interpreter::RuntimeShape::dims(), onert_micro::core::OMRuntimeShape::dims(), onert_micro::core::SplitParams::num_outputs, onert_micro::Ok, onert_micro::core::SplitParams::output_data, and output_shape.
|
inline |
Definition at line 34 of file PALSqrtCommon.h.
References onert_micro::core::OMRuntimeShape::flatSize(), onert_micro::Ok, output_shape, and onert_micro::UnknownError.
|
inline |
Definition at line 34 of file PALSquareCommon.h.
References onert_micro::core::OMRuntimeShape::flatSize(), onert_micro::Ok, output_shape, and onert_micro::UnknownError.
OMStatus onert_micro::execute::pal::SquaredDifference | ( | const core::BinaryArithmeticBroadcastParams & | params, |
const int | flat_size, | ||
const T * | input1_data, | ||
const T * | input2_data, | ||
T * | output_data | ||
) |
Definition at line 29 of file PALSquaredDifferenceCommon.h.
References onert_micro::Ok.
OMStatus onert_micro::execute::pal::StridedSlice | ( | core::StridedSliceParams & | op_params, |
const core::OMRuntimeShape & | unextended_input_shape, | ||
const T * | input_data, | ||
T * | output_data | ||
) |
Definition at line 206 of file PALStridedSlice.h.
References onert_micro::core::OMRuntimeShape::dims(), onert_micro::core::OMRuntimeShape::extendedShape(), onert_micro::Ok, and onert_micro::core::StridedSliceParams::strides.
OMStatus onert_micro::execute::pal::Sub | ( | const core::ArithmeticQuantParams & | params, |
const uint32_t | flat_size, | ||
const int8_t * | input1_data, | ||
const int8_t * | input2_data, | ||
int8_t * | output_data | ||
) |
Definition at line 31 of file PALSub.h.
References ElementWise(), onert_micro::Ok, and SubFunc().
OMStatus onert_micro::execute::pal::Sub | ( | const core::BinaryArithmeticBroadcastParams & | params, |
const int | flat_size, | ||
const T * | input1_data, | ||
const T * | input2_data, | ||
T * | output_data | ||
) |
Definition at line 49 of file PALSubCommon.h.
References onert_micro::Ok.
int8_t onert_micro::execute::pal::SubFunc | ( | int8_t | x, |
int8_t | y, | ||
const core::ArithmeticQuantParams & | params | ||
) |
Definition at line 29 of file PALSubCommon.h.
References onert_micro::core::ArithmeticQuantParams::input1_multiplier, onert_micro::core::ArithmeticQuantParams::input1_offset, onert_micro::core::ArithmeticQuantParams::input1_shift, onert_micro::core::ArithmeticQuantParams::input2_multiplier, onert_micro::core::ArithmeticQuantParams::input2_offset, onert_micro::core::ArithmeticQuantParams::input2_shift, onert_micro::core::ArithmeticQuantParams::left_shift, multiplyByQuantizedMultiplierSmallerThanOneExp(), onert_micro::core::ArithmeticQuantParams::output_multiplier, onert_micro::core::ArithmeticQuantParams::output_offset, onert_micro::core::ArithmeticQuantParams::output_shift, onert_micro::core::ArithmeticQuantParams::quantized_activation_max, and onert_micro::core::ArithmeticQuantParams::quantized_activation_min.
Referenced by BroadcastSub4DSlow(), and Sub().
|
inline |
Definition at line 130 of file ProcessBroadcastShapes.h.
References onert_micro::execute::pal::NdArrayDesc< N >::strides.
Referenced by BroadcastArithmeticOp4DSlow(), BroadcastBinaryOp4DSlow(), BroadcastComparison4DSlowNoScaling(), BroadcastComparison4DSlowWithScaling(), and TransposeImpl().
|
inline |
Definition at line 135 of file ProcessBroadcastShapes.h.
References onert_micro::execute::pal::NdArrayDesc< N >::strides.
OMStatus onert_micro::execute::pal::SVDF | ( | const core::SVDFQuantParams & | params, |
const int8_t * | input_data, | ||
const int8_t * | weights_feature_data, | ||
const int8_t * | weights_time_data, | ||
const int32_t * | bias_data, | ||
int8_t * | state_data, | ||
int8_t * | output_data, | ||
const core::OMRuntimeShape & | input_shape, | ||
const core::OMRuntimeShape & | weights_feature_shape, | ||
const core::OMRuntimeShape & | weights_time_shape, | ||
const core::OMRuntimeShape & | bias_shape, | ||
const core::OMRuntimeShape & | output_shape | ||
) |
Definition at line 35 of file PALSVDF.h.
References onert_micro::core::memory::OMMemoryManager::allocateMemory(), onert_micro::core::memory::OMMemoryManager::deallocateMemory(), luci_interpreter::RuntimeShape::dims(), onert_micro::core::OMRuntimeShape::dims(), onert_micro::core::SVDFQuantParams::effective_scale_1_a, onert_micro::core::SVDFQuantParams::effective_scale_1_b, onert_micro::core::SVDFQuantParams::effective_scale_2_a, onert_micro::core::SVDFQuantParams::effective_scale_2_b, onert_micro::core::SVDFQuantParams::input_zero_point, onert_micro::Ok, output_shape, onert_micro::core::SVDFQuantParams::output_zero_point, and onert_micro::core::SVDFQuantParams::rank.
OMStatus onert_micro::execute::pal::SVDF | ( | const float * | input_data, |
const float * | weights_feature_data, | ||
const float * | weights_time_data, | ||
const float * | bias_data, | ||
float * | state_data, | ||
float * | scratch_data, | ||
float * | output_data, | ||
const int | rank, | ||
const int | input_size, | ||
const int | batch_size, | ||
const int | num_filters, | ||
const int | num_units, | ||
const int | memory_size, | ||
const circle::ActivationFunctionType | activation | ||
) |
Definition at line 138 of file PALSVDFCommon.h.
References onert_micro::Ok.
|
inline |
Definition at line 31 of file PALTanhCommon.h.
References onert_micro::core::OMRuntimeShape::flatSize(), onert_micro::Ok, output_shape, and onert_micro::UnknownError.
|
inline |
Definition at line 78 of file PALTransposeCommon.h.
References onert_micro::UnknownError.
|
inline |
Definition at line 35 of file PALTransposeConvCommon.h.
|
inline |
Definition at line 42 of file PALTransposeConvCommon.h.
References onert_micro::core::FloatConv2D::activation_max, onert_micro::core::FloatConv2D::activation_min, activationFunctionWithMinMax(), luci_interpreter::RuntimeShape::dims(), onert_micro::core::OMRuntimeShape::dims(), luci_interpreter::RuntimeShape::dimsData(), onert_micro::core::OMRuntimeShape::dimsData(), luci_interpreter::RuntimeShape::flatSize(), offset(), onert_micro::Ok, output_shape, onert_micro::core::FloatConv2D::pad_h, onert_micro::core::FloatConv2D::pad_w, onert_micro::core::FloatConv2D::stride_h, and onert_micro::core::FloatConv2D::stride_w.
|
inline |
Definition at line 36 of file PALTransposeCommon.h.
References copyDimsToDesc(), onert_micro::core::OMRuntimeShape::dimensionsCount(), onert_micro::core::OMRuntimeShape::extendedShape(), onert_micro::execute::pal::NdArrayDesc< N >::extents, onert_micro::Ok, onert_micro::core::TransposeParams::perm, onert_micro::execute::pal::NdArrayDesc< N >::strides, and subscriptToIndex().
|
inline |
Definition at line 40 of file PALUnaryOpCommon.h.
References onert_micro::core::OMRuntimeShape::flatSize(), onert_micro::Ok, output_shape, and onert_micro::UnknownError.
OMStatus onert_micro::execute::pal::Unpack | ( | const core::SplitParams & | params, |
const core::OMRuntimeShape & | input_shape, | ||
const T * | input_data, | ||
const core::OMRuntimeShape & | output_shape, | ||
int32_t | axis_value | ||
) |
Definition at line 31 of file PALUnpack.h.
References luci_interpreter::RuntimeShape::dimensionsCount(), onert_micro::core::OMRuntimeShape::dimensionsCount(), luci_interpreter::RuntimeShape::dims(), onert_micro::core::OMRuntimeShape::dims(), onert_micro::core::SplitParams::num_outputs, onert_micro::Ok, onert_micro::core::SplitParams::output_data, and output_shape.
|
constexpr |
Definition at line 32 of file PALGatherND.h.
Referenced by GatherND().