nrf_axon_nn_op_extensions.h

Go to the documentation of this file.

/*
 * Copyright (c) 2025 Nordic Semiconductor ASA
 *
 * SPDX-License-Identifier: LicenseRef-Nordic-5-Clause
 */
 
#pragma once
 
#ifdef __cplusplus
extern "C" {
#endif
#include <stdint.h>
#include <stdbool.h>
#include "nrf_axon_driver.h"
 
typedef nrf_axon_result_e
        (*axon_op_extension_func)
                (uint16_t arg_size, NRF_AXON_PLATFORM_BITWIDTH_UNSIGNED_TYPE *args);
 
typedef struct {
        int8_t *input;
        int8_t *output;
} nrf_axon_nn_op_extension_base1_ptr_args_s;
typedef struct {
        union {
                NRF_AXON_PLATFORM_BITWIDTH_UNSIGNED_TYPE spacer0;
                uint32_t output_multiplier;
        };
        union {
                NRF_AXON_PLATFORM_BITWIDTH_UNSIGNED_TYPE spacer1;
                struct {
                        uint16_t height;
                        uint16_t width;
                };
        };
        union {
                NRF_AXON_PLATFORM_BITWIDTH_UNSIGNED_TYPE spacer2;
                struct {
                        uint16_t channel_cnt;
                        uint8_t output_rounding;
                        int8_t output_zeropoint;
                };
        };
        union {
                NRF_AXON_PLATFORM_BITWIDTH_UNSIGNED_TYPE spacer3;
                struct {
                        uint8_t output_bytewidth;
                        bool quantization_enabled;
                        bool input_is_packed;
                };
        };
} nrf_axon_nn_op_extension_base1_remaining_args_s;
 
typedef struct  {
        nrf_axon_nn_op_extension_base1_ptr_args_s ptr_args;
        nrf_axon_nn_op_extension_base1_remaining_args_s remaining_args;
} nrf_axon_nn_op_extension_base1_args_s;
 
 
typedef struct  {
        struct {
                int8_t *input;
                int8_t *output;
        } ptr_args;
        struct {
                union {
                        NRF_AXON_PLATFORM_BITWIDTH_UNSIGNED_TYPE spacer0;
                        struct {
                                /*< height in elements of the input */
                                uint16_t input_height;
                                /*< width in elements of the input */
                                uint16_t input_width;
                        };
                };
                union {
                        NRF_AXON_PLATFORM_BITWIDTH_UNSIGNED_TYPE spacer1;
                        struct {
                                /*< height in elements of the output */
                                uint16_t output_height;
                                /*< width in elements of the output */
                                uint16_t output_width;
                        };
                };
                union {
                        NRF_AXON_PLATFORM_BITWIDTH_UNSIGNED_TYPE spacer2;
                        struct {
                                /*< channel count in elements of the input */
                                uint16_t input_channel_cnt;
                                /*< channel count in elements of the output */
                                uint16_t output_channel_cnt;
                        };
                };
                union {
                        NRF_AXON_PLATFORM_BITWIDTH_UNSIGNED_TYPE spacer3;
                        struct {
                                uint16_t input_stride;
                        };
                };
        } remaining_args;
} nrf_axon_nn_op_extension_base2_args_s;
 
 
nrf_axon_result_e nrf_axon_nn_op_extension_softmax(
        uint16_t argc, NRF_AXON_PLATFORM_BITWIDTH_UNSIGNED_TYPE *args);
 
nrf_axon_result_e nrf_axon_nn_op_extension_sigmoid(
        uint16_t argc, NRF_AXON_PLATFORM_BITWIDTH_UNSIGNED_TYPE *args);
 
#define NRF_AXON_SIG_TANH_INPUT16_RADIX  12
 
#define NRF_AXON_SIG_TANH_OUTPUT16_RADIX  14
void nrf_axon_compiler_sig_tanh_dequant_consts(int32_t *multiplier, uint8_t *shift);
 
typedef struct  {
        struct {
                nrf_axon_nn_op_extension_base1_ptr_args_s base1;
                int8_t *scratch;
        } ptr_args;
        struct {
                nrf_axon_nn_op_extension_base1_remaining_args_s base1;
                union {
                        NRF_AXON_PLATFORM_BITWIDTH_UNSIGNED_TYPE spacer4;
                        uint32_t input_dequant_multiplier;
                };
                union {
                        NRF_AXON_PLATFORM_BITWIDTH_UNSIGNED_TYPE spacer5;
                        struct {
                                int8_t input_dequant_zero_pt;
                                uint8_t input_dequant_rounding_bits;
 
                                uint8_t input_byte_width;
                        };
                };
        } remaining_args;
} nrf_axon_nn_op_extension_sig_tanh_args_s;
 
nrf_axon_result_e nrf_axon_nn_op_extension_sigmoid_v2(
        uint16_t argc, NRF_AXON_PLATFORM_BITWIDTH_UNSIGNED_TYPE *args);
 
nrf_axon_result_e nrf_axon_nn_op_extension_sigmoid_dequantize_input(uint16_t argc,
        NRF_AXON_PLATFORM_BITWIDTH_UNSIGNED_TYPE *args);
 
nrf_axon_result_e nrf_axon_nn_op_extension_tanh(
        uint16_t argc, NRF_AXON_PLATFORM_BITWIDTH_UNSIGNED_TYPE *args);
 
nrf_axon_result_e nrf_axon_nn_op_extension_tanh_v2(
        uint16_t argc, NRF_AXON_PLATFORM_BITWIDTH_UNSIGNED_TYPE *args);
 
nrf_axon_result_e nrf_axon_nn_op_extension_tanh_dequantize_input(uint16_t argc,
        NRF_AXON_PLATFORM_BITWIDTH_UNSIGNED_TYPE *args);
 
nrf_axon_result_e nrf_axon_nn_op_extension_reshape(
        uint16_t argc, NRF_AXON_PLATFORM_BITWIDTH_UNSIGNED_TYPE *args);
 
nrf_axon_result_e nrf_axon_nn_op_extension_reshape_from_axon(
        uint16_t argc,  NRF_AXON_PLATFORM_BITWIDTH_UNSIGNED_TYPE *args);
 
typedef struct  {
        struct {
                int8_t *input;
                int8_t *output;
        } ptr_args;
        struct {
                union {
                        NRF_AXON_PLATFORM_BITWIDTH_UNSIGNED_TYPE spacer0;
                        struct {
                                uint16_t input_height;
                                uint16_t input_width;
                        };
                };
                union {
                        NRF_AXON_PLATFORM_BITWIDTH_UNSIGNED_TYPE spacer1;
                        struct {
                                uint16_t output_height;
                                uint16_t output_width;
                        };
                };
                union {
                        NRF_AXON_PLATFORM_BITWIDTH_UNSIGNED_TYPE spacer2;
                        struct {
                                uint16_t input_channel_cnt;
                                uint16_t output_channel_cnt;
                        };
                };
                union {
                        NRF_AXON_PLATFORM_BITWIDTH_UNSIGNED_TYPE spacer3;
                        struct {
                                uint16_t input_stride;
                                bool align_corners;
                                bool half_pixel_centers;
                        };
                };
        } remaining_args;
} nrf_axon_nn_op_extension_resize_nearest_neighbor_args_s;
 
nrf_axon_result_e nrf_axon_nn_op_extension_resize_nearest_neighbor(
        uint16_t argc, NRF_AXON_PLATFORM_BITWIDTH_UNSIGNED_TYPE *args);
 
 
 
typedef struct  {
        struct {
                int8_t *space;
                int8_t *batch;
        } ptr_args;
        struct {
                union {
                        NRF_AXON_PLATFORM_BITWIDTH_UNSIGNED_TYPE spacer0;
                        struct {
                                uint16_t space_height;
                                uint16_t space_width;
                        };
                };
                union {
                        NRF_AXON_PLATFORM_BITWIDTH_UNSIGNED_TYPE spacer1;
                        struct {
                                uint16_t batch_height;
                                uint16_t batch_width;
                        };
                };
                union {
                        NRF_AXON_PLATFORM_BITWIDTH_UNSIGNED_TYPE spacer2;
                        struct {
                                uint16_t channel_cnt;
                                uint16_t zero_point;
                        };
                };
                union {
                        NRF_AXON_PLATFORM_BITWIDTH_UNSIGNED_TYPE spacer3;
                        struct {
                                uint16_t input_stride;
                                uint16_t batch_cnt;
                        };
                };
                union {
                        NRF_AXON_PLATFORM_BITWIDTH_UNSIGNED_TYPE spacer4;
                        struct {
                                uint8_t pad_top;
                                uint8_t pad_bottom;
                                uint8_t pad_left;
                                uint8_t pad_right;
                        };
                };
                union {
                        NRF_AXON_PLATFORM_BITWIDTH_UNSIGNED_TYPE space5;
                        struct {
                                uint16_t block_height;
                                uint16_t block_width;
                        };
                };
        } remaining_args;
} nrf_axon_nn_op_extension_space_to_from_batch_args_s;
 
nrf_axon_result_e nrf_axon_nn_cpu_op_space_to_batch(
        uint16_t argc, NRF_AXON_PLATFORM_BITWIDTH_UNSIGNED_TYPE *args);
 
nrf_axon_result_e nrf_axon_nn_cpu_op_batch_to_space(
        uint16_t argc, NRF_AXON_PLATFORM_BITWIDTH_UNSIGNED_TYPE *args);
 
#ifdef __cplusplus
} /* extern "C" { */
#endif