Data movement
Pad, reshape, transpose, concatenate, gather, resize, strided slice, tile, broadcast, scatter, mirror pad, and sequence/slice-update utilities. 77 functions, declared in 7 modules.
| Function | Summary | Module |
|---|---|---|
arm_batch_to_space_nd_s16 |
Batch to Space ND function for s16 data type. | arm_nnfunctions.h |
arm_batch_to_space_nd_s8 |
Batch to Space ND function for s8 data type. | arm_nnfunctions.h |
arm_broadcast_to_s16 |
Broadcast an int16 tensor to a target shape. | arm_nnfunctions.h |
arm_broadcast_to_s8 |
Broadcast an int8 tensor to a target shape. | arm_nnfunctions.h |
arm_concatenation_f16 |
Concatenate float32 tensors of any rank along one axis. | NNSupport |
arm_concatenation_f16_w |
Concatenate tensors along the W axis. | NNSupport |
arm_concatenation_f16_x |
Concatenate tensors along the X axis. | NNSupport |
arm_concatenation_f16_y |
Concatenate tensors along the Y axis. | NNSupport |
arm_concatenation_f16_z |
Concatenate tensors along the Z axis. | NNSupport |
arm_concatenation_f32 |
Concatenate float32 tensors of any rank along one axis. | NNSupport |
arm_concatenation_f32_w |
Concatenate tensors along the W axis. | NNSupport |
arm_concatenation_f32_x |
Concatenate tensors along the X axis. | NNSupport |
arm_concatenation_f32_y |
Concatenate tensors along the Y axis. | NNSupport |
arm_concatenation_f32_z |
Concatenate tensors along the Z axis. | NNSupport |
arm_concatenation_s16 |
int16/uint16 concatenation function to be used for concatenating N-tensors along the target axis | arm_nnfunctions.h |
arm_concatenation_s32 |
int32/uint32 concatenation function to be used for concatenating N-tensors along the target axis | arm_nnfunctions.h |
arm_concatenation_s8 |
int8/uint8 concatenation function to be used for concatenating N-tensors along the target axis | arm_nnfunctions.h |
arm_concatenation_s8_w |
int8/uint8 concatenation function to be used for concatenating N-tensors along the W axis (Batch size) This function should be called for each input tensor to… | arm_nnfunctions.h |
arm_concatenation_s8_x |
int8/uint8 concatenation function to be used for concatenating N-tensors along the X axis This function should be called for each input tensor to concatenate. | arm_nnfunctions.h |
arm_concatenation_s8_y |
int8/uint8 concatenation function to be used for concatenating N-tensors along the Y axis This function should be called for each input tensor to concatenate. | arm_nnfunctions.h |
arm_concatenation_s8_z |
int8/uint8 concatenation function to be used for concatenating N-tensors along the Z axis This function should be called for each input tensor to concatenate. | arm_nnfunctions.h |
arm_depth_to_space_s16 |
Depth to Space function for s16 data type. | arm_nnfunctions.h |
arm_depth_to_space_s8 |
Depth to Space function for s8 data type. | arm_nnfunctions.h |
arm_dynamic_update_slice_s16 |
Update a slice of an int16 operand tensor at runtime-determined indices. | arm_nnfunctions.h |
arm_dynamic_update_slice_s8 |
Update a slice of an int8 operand tensor at runtime-determined indices. | arm_nnfunctions.h |
arm_gather_f16 |
Gather contiguous slices along an axis. | Gather Functions: |
arm_gather_f32 |
Gather contiguous slices along an axis. | Gather Functions: |
arm_gather_nd_f16 |
Gather contiguous slices using coordinate tuples. | Gather Functions: |
arm_gather_nd_f32 |
Gather contiguous slices using coordinate tuples. | Gather Functions: |
arm_gather_nd_s16 |
Gathernd slices for int16 tensors. | arm_nnfunctions.h |
arm_gather_nd_s8 |
Gathernd slices for int8 tensors. | arm_nnfunctions.h |
arm_gather_s16 |
Gather elements along an axis for int16 tensors. | arm_nnfunctions.h |
arm_gather_s8 |
Gather elements along an axis for int8 tensors. | arm_nnfunctions.h |
arm_mirror_pad_s16 |
Mirror-pad an int16 tensor (REFLECT or SYMMETRIC mode). | arm_nnfunctions.h |
arm_mirror_pad_s8 |
Mirror-pad an int8 tensor (REFLECT or SYMMETRIC mode). | arm_nnfunctions.h |
arm_nn_fill_f16 |
Fill a float16 vector with one value; bit copy of value, NaN payload included. | Elementwise Functions |
arm_nn_fill_f32 |
Fill a float32 vector with one value. | Elementwise Functions |
arm_pack_f16 |
Stack float32 tensors of equal shape along a new axis (TFLite PACK). | NNSupport |
arm_pack_f32 |
Stack float32 tensors of equal shape along a new axis (TFLite PACK). | NNSupport |
arm_pad_f16 |
Pad a tensor with a constant value. | Pad Layer Functions: |
arm_pad_f32 |
Pad a tensor with a constant value. | Pad Layer Functions: |
arm_pad_s16 |
Expands the size of the input by adding constant values before and after the data, in all dimensions. | arm_nnfunctions.h |
arm_pad_s8 |
Expands the size of the input by adding constant values before and after the data, in all dimensions. | arm_nnfunctions.h |
arm_reshape_f16 |
Reshape by copying data without changing element order. | NNSupport |
arm_reshape_f32 |
Reshape by copying data without changing element order. | NNSupport |
arm_reshape_s8 |
Reshape a s8 vector into another with different shape. | arm_nnfunctions.h |
arm_resize_nearest_neighbor_f16 |
Nearest-neighbor resize of a float32 NHWC tensor. | Reshape Functions |
arm_resize_nearest_neighbor_f16_get_buffer_size |
Scratch size in bytes for armresizenearestneighborf32() / armresizenearestneighborf16(). | Reshape Functions |
arm_resize_nearest_neighbor_f32 |
Nearest-neighbor resize of a float32 NHWC tensor. | Reshape Functions |
arm_resize_nearest_neighbor_f32_get_buffer_size |
Scratch size in bytes for armresizenearestneighborf32() / armresizenearestneighborf16(). | Reshape Functions |
arm_resize_nearest_neighbor_s16 |
Nearest neighbor resize function for s16 data. | arm_nnfunctions.h |
arm_resize_nearest_neighbor_s8 |
Nearest neighbor resize function for s8 data. | arm_nnfunctions.h |
arm_reverse_sequence_s16 |
Reverse variable-length sequences along a dimension for int16. | arm_nnfunctions.h |
arm_reverse_sequence_s8 |
Reverse variable-length sequences along a dimension for int8. | arm_nnfunctions.h |
arm_scatter_nd_s16 |
Scatter updates into a zero-initialized output tensor for int16. | arm_nnfunctions.h |
arm_scatter_nd_s8 |
Scatter updates into a zero-initialized output tensor for int8. | arm_nnfunctions.h |
arm_space_to_batch_nd_s16 |
Space to Batch ND function for s16 data type. | arm_nnfunctions.h |
arm_space_to_batch_nd_s8 |
Space to Batch ND function for s8 data type. | arm_nnfunctions.h |
arm_space_to_depth_s16 |
Space to Depth function for s16 data type. | arm_nnfunctions.h |
arm_space_to_depth_s8 |
Space to Depth function for s8 data type. | arm_nnfunctions.h |
arm_split_f16 |
Split a float32 tensor of any rank into several tensors along one axis. | Elementwise Functions |
arm_split_f32 |
Split a float32 tensor of any rank into several tensors along one axis. | NNSupport |
arm_split_s16 |
int16/uint16 split function to be used for splitting a tensor into multiple tensors along the target axis | arm_nnfunctions.h |
arm_split_s8 |
int8/uint8 split function to be used for splitting a tensor into multiple tensors along the target axis | arm_nnfunctions.h |
arm_strided_slice_f16 |
Strided slice for float32 data (pure copy, TensorFlow Lite compatible). | Elementwise Functions |
arm_strided_slice_f32 |
Strided slice for float32 data (pure copy, TensorFlow Lite compatible). | Slicing Functions: |
arm_strided_slice_s16 |
Strided slice function for int16 data. | arm_nnfunctions.h |
arm_strided_slice_s32 |
Strided slice function for int32 data. | arm_nnfunctions.h |
arm_strided_slice_s8 |
Strided slice function for int8 data. | arm_nnfunctions.h |
arm_tile_s16 |
Tile an int16 tensor along each dimension. | arm_nnfunctions.h |
arm_tile_s8 |
Tile an int8 tensor along each dimension. | arm_nnfunctions.h |
arm_transpose_f16 |
Transpose a floating-point tensor. | NNSupport |
arm_transpose_f32 |
Transpose a floating-point tensor. | NNSupport |
arm_transpose_s16 |
Basic s16 transpose function. | arm_nnfunctions.h |
arm_transpose_s8 |
Basic transpose function. | arm_nnfunctions.h |
arm_unpack_f16 |
Unstack a float32 tensor along one axis into inputshape[axis] tensors (TFLite UNPACK). | NNSupport |
arm_unpack_f32 |
Unstack a float32 tensor along one axis into inputshape[axis] tensors (TFLite UNPACK). | NNSupport |