arm_nn_axis_copy_common.h

Machine-readable model

• arm_nn_axis_copy_planfunction
• arm_nn_axis_copy_ptrs_okfunction
• ARM_NN_AXIS_COPY_DEFINEmacro
• arm_nn_axis_scatter_f32function
• arm_nn_axis_gather_f32function
• arm_nn_axis_scatter_f16function
• arm_nn_axis_gather_f16function

function

arm_nn_axis_copy_plan

C

Include/Internal/arm_nn_axis_copy_common.h:45

static arm_cmsis_nn_status arm_nn_axis_copy_plan(
    const int32_t *shape,
    const int32_t dims,
    const int32_t axis,
    const int32_t inner_begin,
    const int32_t axis_len,
    const int32_t num,
    const int32_t *sizes,
    int32_t *outer,
    int32_t *inner
)

Parameters

Parameters of arm_nn_axis_copy_plan

Name	Type	Description
shape	const int32_t *
dims	const int32_t
axis	const int32_t
inner_begin	const int32_t
axis_len	const int32_t
num	const int32_t
sizes	const int32_t *
outer	int32_t *
inner	int32_t *

function

arm_nn_axis_copy_ptrs_ok

C

Include/Internal/arm_nn_axis_copy_common.h:123

static int32_t arm_nn_axis_copy_ptrs_ok(const void *const *ptrs, const int32_t num, const int32_t total)

Parameters

Parameters of arm_nn_axis_copy_ptrs_ok

Name	Type	Description
ptrs	const void *const *
num	const int32_t
total	const int32_t

macro

ARM_NN_AXIS_COPY_DEFINE

C

Include/Internal/arm_nn_axis_copy_common.h:145

#define ARM_NN_AXIS_COPY_DEFINE(SUFFIX, TYPE)

Parameters

Parameters of ARM_NN_AXIS_COPY_DEFINE

Name	Description
SUFFIX
TYPE

function

arm_nn_axis_scatter_f32

C

Include/Internal/arm_nn_axis_copy_common.h:198

static void arm_nn_axis_scatter_f32(
    const float32_t *packed,
    const int32_t outer,
    const int32_t num,
    const int32_t *sizes,
    const int32_t inner,
    float32_t *const *slices
)

Parameters

Parameters of arm_nn_axis_scatter_f32

Name	Type	Description
packed	const float32_t *
outer	const int32_t
num	const int32_t
sizes	const int32_t *
inner	const int32_t
slices	float32_t *const *

function

arm_nn_axis_gather_f32

C

Include/Internal/arm_nn_axis_copy_common.h:198

static void arm_nn_axis_gather_f32(
    const float32_t *const *slices,
    const int32_t outer,
    const int32_t num,
    const int32_t *sizes,
    const int32_t inner,
    float32_t *packed
)

Parameters

Parameters of arm_nn_axis_gather_f32

Name	Type	Description
slices	const float32_t *const *
outer	const int32_t
num	const int32_t
sizes	const int32_t *
inner	const int32_t
packed	float32_t *

function

arm_nn_axis_scatter_f16

C

Include/Internal/arm_nn_axis_copy_common.h:201

static void arm_nn_axis_scatter_f16(
    const float16_t *packed,
    const int32_t outer,
    const int32_t num,
    const int32_t *sizes,
    const int32_t inner,
    float16_t *const *slices
)

Parameters

Parameters of arm_nn_axis_scatter_f16

Name	Type	Description
packed	const float16_t *
outer	const int32_t
num	const int32_t
sizes	const int32_t *
inner	const int32_t
slices	float16_t *const *

function

arm_nn_axis_gather_f16

C

Include/Internal/arm_nn_axis_copy_common.h:201

static void arm_nn_axis_gather_f16(
    const float16_t *const *slices,
    const int32_t outer,
    const int32_t num,
    const int32_t *sizes,
    const int32_t inner,
    float16_t *packed
)

Parameters

Parameters of arm_nn_axis_gather_f16

Name	Type	Description
slices	const float16_t *const *
outer	const int32_t
num	const int32_t
sizes	const int32_t *
inner	const int32_t
packed	float16_t *