21.02
|
#include "helpers.h"
Go to the source code of this file.
Functions | |
__kernel void | l2_normalize_x (__global uchar *src_ptr, uint src_stride_x, uint src_step_x, uint src_stride_y, uint src_step_y, uint src_offset_first_element_in_bytes, __global uchar *sum_ptr, uint sum_stride_x, uint sum_step_x, uint sum_stride_y, uint sum_step_y, uint sum_offset_first_element_in_bytes, __global uchar *dst_ptr, uint dst_stride_x, uint dst_step_x, uint dst_stride_y, uint dst_step_y, uint dst_offset_first_element_in_bytes, DATA_TYPE epsilon) |
This kernel performs l2 normalization on x-axis. More... | |
__kernel void | l2_normalize_y (__global uchar *src_ptr, uint src_stride_x, uint src_step_x, uint src_stride_y, uint src_step_y, uint src_offset_first_element_in_bytes, __global uchar *sum_ptr, uint sum_stride_x, uint sum_step_x, uint sum_stride_y, uint sum_step_y, uint sum_offset_first_element_in_bytes, __global uchar *dst_ptr, uint dst_stride_x, uint dst_step_x, uint dst_stride_y, uint dst_step_y, uint dst_offset_first_element_in_bytes, DATA_TYPE epsilon) |
This kernel performs l2 normalization on y-axis. More... | |
__kernel void | l2_normalize_z (__global uchar *src_ptr, uint src_stride_x, uint src_step_x, uint src_stride_y, uint src_step_y, uint src_stride_z, uint src_step_z, uint src_offset_first_element_in_bytes, __global uchar *sum_ptr, uint sum_stride_x, uint sum_step_x, uint sum_stride_y, uint sum_step_y, uint sum_stride_z, uint sum_step_z, uint sum_offset_first_element_in_bytes, __global uchar *dst_ptr, uint dst_stride_x, uint dst_step_x, uint dst_stride_y, uint dst_step_y, uint dst_stride_z, uint dst_step_z, uint dst_offset_first_element_in_bytes, DATA_TYPE epsilon) |
This kernel performs l2 normalization on z-axis. More... | |
__kernel void l2_normalize_x | ( | __global uchar * | src_ptr, |
uint | src_stride_x, | ||
uint | src_step_x, | ||
uint | src_stride_y, | ||
uint | src_step_y, | ||
uint | src_offset_first_element_in_bytes, | ||
__global uchar * | sum_ptr, | ||
uint | sum_stride_x, | ||
uint | sum_step_x, | ||
uint | sum_stride_y, | ||
uint | sum_step_y, | ||
uint | sum_offset_first_element_in_bytes, | ||
__global uchar * | dst_ptr, | ||
uint | dst_stride_x, | ||
uint | dst_step_x, | ||
uint | dst_stride_y, | ||
uint | dst_step_y, | ||
uint | dst_offset_first_element_in_bytes, | ||
DATA_TYPE | epsilon | ||
) |
This kernel performs l2 normalization on x-axis.
[in] | src_ptr | Pointer to the source tensor. Supported data types: F16/F32 |
[in] | src_stride_x | Stride of the source tensor in X dimension (in bytes) |
[in] | src_step_x | src_stride_x * number of elements along X processed per workitem(in bytes) |
[in] | src_stride_y | Stride of the source tensor in Y dimension (in bytes) |
[in] | src_step_y | src_stride_y * number of elements along X processed per workitem(in bytes) |
[in] | src_offset_first_element_in_bytes | The offset of the first element in the source tensor |
[in] | sum_ptr | Pointer to the source tensor. Supported data types: F16/F32 |
[in] | sum_stride_x | Stride of the source tensor in X dimension (in bytes) |
[in] | sum_step_x | sum_stride_x * number of elements along X processed per workitem(in bytes) |
[in] | sum_stride_y | Stride of the source tensor in Y dimension (in bytes) |
[in] | sum_step_y | sum_stride_y * number of elements along Y processed per workitem(in bytes) |
[in] | sum_offset_first_element_in_bytes | The offset of the first element in the source tensor |
[out] | dst_ptr | Pointer to the destination tensor. Supported data types: same as src_ptr |
[in] | dst_stride_x | Stride of the destination tensor in X dimension (in bytes) |
[in] | dst_step_x | dst_stride_x * number of elements along X processed per workitem(in bytes) |
[in] | dst_stride_y | Stride of the destination tensor in Y dimension (in bytes) |
[in] | dst_step_y | dst_stride_y * number of elements along Y processed per workitem(in bytes) |
[in] | dst_offset_first_element_in_bytes | The offset of the first element in the destination tensor |
[in] | epsilon | Epsilon value |
Definition at line 51 of file l2_normalize.cl.
References CONVERT_TO_IMAGE_STRUCT, DATA_TYPE, arm_compute::test::validation::dst, Image::ptr, arm_compute::test::validation::src, sum(), and VEC_DATA_TYPE.
__kernel void l2_normalize_y | ( | __global uchar * | src_ptr, |
uint | src_stride_x, | ||
uint | src_step_x, | ||
uint | src_stride_y, | ||
uint | src_step_y, | ||
uint | src_offset_first_element_in_bytes, | ||
__global uchar * | sum_ptr, | ||
uint | sum_stride_x, | ||
uint | sum_step_x, | ||
uint | sum_stride_y, | ||
uint | sum_step_y, | ||
uint | sum_offset_first_element_in_bytes, | ||
__global uchar * | dst_ptr, | ||
uint | dst_stride_x, | ||
uint | dst_step_x, | ||
uint | dst_stride_y, | ||
uint | dst_step_y, | ||
uint | dst_offset_first_element_in_bytes, | ||
DATA_TYPE | epsilon | ||
) |
This kernel performs l2 normalization on y-axis.
[in] | src_ptr | Pointer to the source tensor. Supported data types: F16/F32 |
[in] | src_stride_x | Stride of the source tensor in X dimension (in bytes) |
[in] | src_step_x | src_stride_x * number of elements along Y processed per workitem(in bytes) |
[in] | src_stride_y | Stride of the source tensor in Y dimension (in bytes) |
[in] | src_step_y | src_stride_y * number of elements along X processed per workitem(in bytes) |
[in] | src_offset_first_element_in_bytes | The offset of the first element in the source tensor |
[in] | sum_ptr | Pointer to the source tensor. Supported data types: F16/F32 |
[in] | sum_stride_x | Stride of the source tensor in X dimension (in bytes) |
[in] | sum_step_x | sum_stride_x * number of elements along X processed per workitem(in bytes) |
[in] | sum_stride_y | Stride of the source tensor in Y dimension (in bytes) |
[in] | sum_step_y | sum_stride_y * number of elements along Y processed per workitem(in bytes) |
[in] | sum_offset_first_element_in_bytes | The offset of the first element in the source tensor |
[out] | dst_ptr | Pointer to the destination tensor. Supported data types: same as src_ptr |
[in] | dst_stride_x | Stride of the destination tensor in X dimension (in bytes) |
[in] | dst_step_x | dst_stride_x * number of elements along X processed per workitem(in bytes) |
[in] | dst_stride_y | Stride of the destination tensor in Y dimension (in bytes) |
[in] | dst_step_y | dst_stride_y * number of elements along Y processed per workitem(in bytes) |
[in] | dst_offset_first_element_in_bytes | The offset of the first element in the destination tensor |
[in] | epsilon | Epsilon value |
Definition at line 94 of file l2_normalize.cl.
References CONVERT_TO_IMAGE_STRUCT, DATA_TYPE, arm_compute::test::validation::dst, Image::ptr, arm_compute::test::validation::src, sum(), and VEC_DATA_TYPE.
__kernel void l2_normalize_z | ( | __global uchar * | src_ptr, |
uint | src_stride_x, | ||
uint | src_step_x, | ||
uint | src_stride_y, | ||
uint | src_step_y, | ||
uint | src_stride_z, | ||
uint | src_step_z, | ||
uint | src_offset_first_element_in_bytes, | ||
__global uchar * | sum_ptr, | ||
uint | sum_stride_x, | ||
uint | sum_step_x, | ||
uint | sum_stride_y, | ||
uint | sum_step_y, | ||
uint | sum_stride_z, | ||
uint | sum_step_z, | ||
uint | sum_offset_first_element_in_bytes, | ||
__global uchar * | dst_ptr, | ||
uint | dst_stride_x, | ||
uint | dst_step_x, | ||
uint | dst_stride_y, | ||
uint | dst_step_y, | ||
uint | dst_stride_z, | ||
uint | dst_step_z, | ||
uint | dst_offset_first_element_in_bytes, | ||
DATA_TYPE | epsilon | ||
) |
This kernel performs l2 normalization on z-axis.
[in] | src_ptr | Pointer to the source tensor. Supported data types: F16/F32 |
[in] | src_stride_x | Stride of the source tensor in X dimension (in bytes) |
[in] | src_step_x | src_stride_x * number of elements along Y processed per workitem(in bytes) |
[in] | src_stride_y | Stride of the source tensor in Y dimension (in bytes) |
[in] | src_step_y | src_stride_y * number of elements along X processed per workitem(in bytes) |
[in] | src_stride_z | Stride of the source tensor in Z dimension (in bytes) |
[in] | src_step_z | src_stride_z * number of elements along Z processed per workitem(in bytes) |
[in] | src_offset_first_element_in_bytes | The offset of the first element in the source tensor |
[in] | sum_ptr | Pointer to the source tensor. Supported data types: F16/F32 |
[in] | sum_stride_x | Stride of the source tensor in X dimension (in bytes) |
[in] | sum_step_x | sum_stride_x * number of elements along X processed per workitem(in bytes) |
[in] | sum_stride_y | Stride of the source tensor in Y dimension (in bytes) |
[in] | sum_step_y | sum_stride_y * number of elements along Y processed per workitem(in bytes) |
[in] | sum_stride_z | Stride of the source tensor in Z dimension (in bytes) |
[in] | sum_step_z | sum_stride_z * number of elements along Z processed per workitem(in bytes) |
[in] | sum_offset_first_element_in_bytes | The offset of the first element in the source tensor |
[out] | dst_ptr | Pointer to the destination tensor. Supported data types: same as src_ptr |
[in] | dst_stride_x | Stride of the destination tensor in X dimension (in bytes) |
[in] | dst_step_x | dst_stride_x * number of elements along X processed per workitem(in bytes) |
[in] | dst_stride_y | Stride of the destination tensor in Y dimension (in bytes) |
[in] | dst_step_y | dst_stride_y * number of elements along Y processed per workitem(in bytes) |
[in] | dst_stride_z | Stride of the destination tensor in Z dimension (in bytes) |
[in] | dst_step_z | dst_stride_z * number of elements along Y processed per workitem(in bytes) |
[in] | dst_offset_first_element_in_bytes | The offset of the first element in the destination tensor |
[in] | epsilon | Epsilon value |
Definition at line 145 of file l2_normalize.cl.
References CONVERT_TO_TENSOR3D_STRUCT, DATA_TYPE, arm_compute::test::validation::dst, Tensor3D::ptr, arm_compute::test::validation::src, sum(), and VEC_DATA_TYPE.