nrf_axon_nn_op_extensions.h File Reference

#include <stdint.h>
#include <stdbool.h>
#include "nrf_axon_driver.h"

Go to the source code of this file.

Data Structures
struct	nrf_axon_nn_op_extension_base1_ptr_args_s
	Structure compiled into the command buffer then passed to some CPU op extension functions. More...
struct	nrf_axon_nn_op_extension_base1_remaining_args_s
struct	nrf_axon_nn_op_extension_base1_args_s
struct	nrf_axon_nn_op_extension_base2_args_s
	Structure compiled into the command buffer then passed to some CPU op extension functions. More...
struct	nrf_axon_nn_op_extension_sig_tanh_args_s
	Structure compiled into the command buffer then passed to some CPU op extension functions. More...
struct	nrf_axon_nn_op_extension_resize_nearest_neighbor_args_s
	Structure compiled into the command buffer then passed to the CPU op extension function resize_nearest_neighbor. More...
struct	nrf_axon_nn_op_extension_space_to_from_batch_args_s
	Structure compiled into the command buffer then passed to the CPU op extension functions space_to_batcnh and batch_to_space. More...

Macros
#define	NRF_AXON_SIG_TANH_INPUT16_RADIX   12
	radix of 16bit input to sigmoid and tanh
#define	NRF_AXON_SIG_TANH_OUTPUT16_RADIX   14
	radix of 16bit output to sigmoid and tanh

Typedefs
typedef nrf_axon_result_e(*	axon_op_extension_func) (uint16_t arg_size, NRF_AXON_PLATFORM_BITWIDTH_UNSIGNED_TYPE *args)
	Function prototype for all operation extension functions.

Functions
nrf_axon_result_e	nrf_axon_nn_op_extension_softmax (uint16_t argc, NRF_AXON_PLATFORM_BITWIDTH_UNSIGNED_TYPE *args)
	Implements softmax operator.
nrf_axon_result_e	nrf_axon_nn_op_extension_sigmoid (uint16_t argc, NRF_AXON_PLATFORM_BITWIDTH_UNSIGNED_TYPE *args)
	Implements sigmoid operator.
void	nrf_axon_compiler_sig_tanh_dequant_consts (int32_t *multiplier, uint8_t *shift)
	adjusts dequantization multiplier and scaler to produce a q3.12
nrf_axon_result_e	nrf_axon_nn_op_extension_sigmoid_v2 (uint16_t argc, NRF_AXON_PLATFORM_BITWIDTH_UNSIGNED_TYPE *args)
	Implements sigmoid operator.
nrf_axon_result_e	nrf_axon_nn_op_extension_sigmoid_dequantize_input (uint16_t argc, NRF_AXON_PLATFORM_BITWIDTH_UNSIGNED_TYPE *args)
	Implements sigmoid operator.
nrf_axon_result_e	nrf_axon_nn_op_extension_tanh (uint16_t argc, NRF_AXON_PLATFORM_BITWIDTH_UNSIGNED_TYPE *args)
	Implements tanh operator.
nrf_axon_result_e	nrf_axon_nn_op_extension_tanh_v2 (uint16_t argc, NRF_AXON_PLATFORM_BITWIDTH_UNSIGNED_TYPE *args)
	Implements tanh operator.
nrf_axon_result_e	nrf_axon_nn_op_extension_tanh_dequantize_input (uint16_t argc, NRF_AXON_PLATFORM_BITWIDTH_UNSIGNED_TYPE *args)
	Implements tanh operator.
nrf_axon_result_e	nrf_axon_nn_op_extension_reshape (uint16_t argc, NRF_AXON_PLATFORM_BITWIDTH_UNSIGNED_TYPE *args)
	Implements reshape operator.
nrf_axon_result_e	nrf_axon_nn_op_extension_reshape_from_axon (uint16_t argc, NRF_AXON_PLATFORM_BITWIDTH_UNSIGNED_TYPE *args)
nrf_axon_result_e	nrf_axon_nn_op_extension_resize_nearest_neighbor (uint16_t argc, NRF_AXON_PLATFORM_BITWIDTH_UNSIGNED_TYPE *args)
	Implements resize_nearest_neighbor operator.
nrf_axon_result_e	nrf_axon_nn_cpu_op_space_to_batch (uint16_t argc, NRF_AXON_PLATFORM_BITWIDTH_UNSIGNED_TYPE *args)
	Implements space_to_batch operator.
nrf_axon_result_e	nrf_axon_nn_cpu_op_batch_to_space (uint16_t argc, NRF_AXON_PLATFORM_BITWIDTH_UNSIGNED_TYPE *args)
	Implements batch_to_space operator.