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_args_s
	Structure compiled into the command buffer then passed to some CPU op extension functions. More...
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_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...

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.
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_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_reshape (uint16_t argc, NRF_AXON_PLATFORM_BITWIDTH_UNSIGNED_TYPE *args)
	Implements reshape operator.
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.