24.02.1
|
#include "helpers.h"
Go to the source code of this file.
Functions | |
__kernel void | strided_slice (__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_stride_w, uint input_step_w, 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_stride_w, uint output_step_w, uint output_offset_first_element_in_bytes) |
Perform a strided slice operation on a given input. More... | |
__kernel void strided_slice | ( | __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_stride_w, | ||
uint | input_step_w, | ||
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_stride_w, | ||
uint | output_step_w, | ||
uint | output_offset_first_element_in_bytes | ||
) |
Perform a strided slice operation on a given input.
[in] | input_ptr | Pointer to the source tensor. Supported data types: All |
[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_stride_w | Stride of the source tensor in W dimension (in bytes) |
[in] | input_step_w | input_stride_w * number of elements along W 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 destination tensor. Supported data types: same as input_ptr |
[in] | output_stride_x | Stride of the destination 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 destination 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 destination 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_stride_w | Stride of the destination tensor in W dimension (in bytes) |
[in] | output_step_w | output_stride_w * number of elements along W processed per workitem(in bytes) |
[in] | output_offset_first_element_in_bytes | The offset of the first element in the destination tensor |
Definition at line 56 of file slice_ops.cl.
References CONVERT_TO_TENSOR4D_STRUCT, CONVERT_TO_TENSOR4D_STRUCT_NO_STEP, arm_compute::test::validation::input, offset(), Tensor4D::ptr, VEC_DATA_TYPE, VEC_SIZE, VLOAD, and VSTORE.