57 int3
id = (int3)(get_global_id(0), get_global_id(1), get_global_id(2));
60 int linear_idx =
id.x +
id.y * input_shape.x +
id.z * input_shape.x * input_shape.y;
64 out_id.x = linear_idx % output_shape.x;
65 out_id.y = (linear_idx / output_shape.x) % output_shape.y;
66 out_id.z = linear_idx / (output_shape.x * output_shape.y);
#define CONVERT_TO_TENSOR3D_STRUCT(name)
Structure to hold 3D tensor information.
__global const uchar * tensor3D_offset(const Tensor3D *tensor, int x, int y, int z)
Get the pointer position of a Tensor3D.
__kernel void reshape_layer(__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, int2 input_shape, int2 output_shape)
Perform tensor reshape.
#define TENSOR3D_DECLARATION(name)
#define CONVERT_TO_TENSOR3D_STRUCT_NO_STEP(name)
__global uchar * ptr
Pointer to the starting postion of the buffer.