21.02
|
#include "helpers.h"
Go to the source code of this file.
Functions | |
__kernel void | max_unpooling_layer_2 (__global uchar *input_ptr, uint input_stride_x, uint input_step_x, uint input_stride_y, uint input_step_y, uint input_stride_z, uint input_step_z, uint input_offset_first_element_in_bytes, __global uchar *output_ptr, uint output_stride_x, uint output_step_x, uint output_stride_y, uint output_step_y, uint output_stride_z, uint output_step_z, uint output_offset_first_element_in_bytes, __global uchar *indices_ptr, uint indices_stride_x, uint indices_step_x, uint indices_stride_y, uint indices_step_y, uint indices_stride_z, uint indices_step_z, uint indices_offset_first_element_in_bytes) |
Performs max unpooling function with pool size equal to 2. More... | |
__kernel void max_unpooling_layer_2 | ( | __global uchar * | input_ptr, |
uint | input_stride_x, | ||
uint | input_step_x, | ||
uint | input_stride_y, | ||
uint | input_step_y, | ||
uint | input_stride_z, | ||
uint | input_step_z, | ||
uint | input_offset_first_element_in_bytes, | ||
__global uchar * | output_ptr, | ||
uint | output_stride_x, | ||
uint | output_step_x, | ||
uint | output_stride_y, | ||
uint | output_step_y, | ||
uint | output_stride_z, | ||
uint | output_step_z, | ||
uint | output_offset_first_element_in_bytes, | ||
__global uchar * | indices_ptr, | ||
uint | indices_stride_x, | ||
uint | indices_step_x, | ||
uint | indices_stride_y, | ||
uint | indices_step_y, | ||
uint | indices_stride_z, | ||
uint | indices_step_z, | ||
uint | indices_offset_first_element_in_bytes | ||
) |
Performs max unpooling function with pool size equal to 2.
[in] | input_ptr | Pointer to the source tensor. Supported data types: F16/F32 |
[in] | input_stride_x | Stride of the source tensor in X dimension (in bytes) |
[in] | input_step_x | input_stride_x * number of elements along X processed per workitem(in bytes) |
[in] | input_stride_y | Stride of the source tensor in Y dimension (in bytes) |
[in] | input_step_y | input_stride_y * number of elements along Y processed per workitem(in bytes) |
[in] | input_stride_z | Stride of the source tensor in Z dimension (in bytes) |
[in] | input_step_z | input_stride_z * number of elements along Z processed per workitem(in bytes) |
[in] | input_offset_first_element_in_bytes | The offset of the first element in the source tensor |
[out] | output_ptr | Pointer to the output tensor. Supported data types: same as input_ptr |
[in] | output_stride_x | Stride of the output tensor in X dimension (in bytes) |
[in] | output_step_x | output_stride_x * number of elements along X processed per workitem(in bytes) |
[in] | output_stride_y | Stride of the output tensor in Y dimension (in bytes) |
[in] | output_step_y | output_stride_y * number of elements along Y processed per workitem(in bytes) |
[in] | output_stride_z | Stride of the output tensor in Z dimension (in bytes) |
[in] | output_step_z | output_stride_z * number of elements along Z processed per workitem(in bytes) |
[in] | output_offset_first_element_in_bytes | The offset of the first element in the output tensor |
[in] | indices_ptr | Pointer to the indices tensor. Supported data types: U32 |
[in] | indices_stride_x | Stride of the indices tensor in X dimension (in bytes) |
[in] | indices_step_x | indices_stride_x * number of elements along X processed per workitem(in bytes) |
[in] | indices_stride_y | Stride of the indices tensor in Y dimension (in bytes) |
[in] | indices_step_y | indices_stride_y * number of elements along Y processed per workitem(in bytes) |
[in] | indices_stride_z | Stride of the indices tensor in Z dimension (in bytes) |
[in] | indices_step_z | indices_stride_z * number of elements along Z processed per workitem(in bytes) |
[in] | indices_offset_first_element_in_bytes | The offset of the first element in the indices tensor |
Definition at line 59 of file unpooling_layer.cl.
References CONVERT_TO_TENSOR3D_STRUCT, CONVERT_TO_TENSOR3D_STRUCT_NO_UPDATE_PTR, DATA_TYPE, arm_compute::test::validation::input, and Tensor3D::ptr.