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  {
        struct {
                int8_t *input;
                int8_t *output;
        } ptr_args;
        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;
                        };
                };
        } 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);
 
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_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_reshape(
        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