|
| CLDepthwiseWeightsReshapeKernel () |
| Default constructor. More...
|
|
| CLDepthwiseWeightsReshapeKernel (const CLDepthwiseWeightsReshapeKernel &)=delete |
| Prevent instances of this class from being copied (As this class contains pointers) More...
|
|
CLDepthwiseWeightsReshapeKernel & | operator= (const CLDepthwiseWeightsReshapeKernel &)=delete |
| Prevent instances of this class from being copied (As this class contains pointers) More...
|
|
| CLDepthwiseWeightsReshapeKernel (CLDepthwiseWeightsReshapeKernel &&)=default |
| Allow instances of this class to be moved. More...
|
|
CLDepthwiseWeightsReshapeKernel & | operator= (CLDepthwiseWeightsReshapeKernel &&)=default |
| Allow instances of this class to be moved. More...
|
|
void | configure (const ICLTensor *input, ICLTensor *output, const ICLTensor *biases=nullptr) |
| Set the input and output of the kernel. More...
|
|
void | run (const Window &window, cl::CommandQueue &queue) override |
| Enqueue the OpenCL kernel to process the given window on the passed OpenCL command queue. More...
|
|
| ICLKernel () |
| Constructor. More...
|
|
cl::Kernel & | kernel () |
| Returns a reference to the OpenCL kernel of this object. More...
|
|
template<typename T > |
void | add_1D_array_argument (unsigned int &idx, const ICLArray< T > *array, const Strides &strides, unsigned int num_dimensions, const Window &window) |
| Add the passed 1D array's parameters to the object's kernel's arguments starting from the index idx. More...
|
|
void | add_1D_tensor_argument (unsigned int &idx, const ICLTensor *tensor, const Window &window) |
| Add the passed 1D tensor's parameters to the object's kernel's arguments starting from the index idx. More...
|
|
void | add_2D_tensor_argument (unsigned int &idx, const ICLTensor *tensor, const Window &window) |
| Add the passed 2D tensor's parameters to the object's kernel's arguments starting from the index idx. More...
|
|
void | add_3D_tensor_argument (unsigned int &idx, const ICLTensor *tensor, const Window &window) |
| Add the passed 3D tensor's parameters to the object's kernel's arguments starting from the index idx. More...
|
|
void | add_4D_tensor_argument (unsigned int &idx, const ICLTensor *tensor, const Window &window) |
| Add the passed 4D tensor's parameters to the object's kernel's arguments starting from the index idx. More...
|
|
template<typename T > |
void | add_argument (unsigned int &idx, T value) |
| Add the passed parameters to the object's kernel's arguments starting from the index idx. More...
|
|
void | set_lws_hint (const cl::NDRange &lws_hint) |
| Set the Local-Workgroup-Size hint. More...
|
|
cl::NDRange | lws_hint () const |
| Return the Local-Workgroup-Size hint. More...
|
|
const std::string & | config_id () const |
| Get the configuration ID. More...
|
|
void | set_target (GPUTarget target) |
| Set the targeted GPU architecture. More...
|
|
void | set_target (cl::Device &device) |
| Set the targeted GPU architecture according to the CL device. More...
|
|
GPUTarget | get_target () const |
| Get the targeted GPU architecture. More...
|
|
size_t | get_max_workgroup_size () |
| Get the maximum workgroup size for the device the CLKernelLibrary uses. More...
|
|
template<typename T , unsigned int dimension_size> |
void | add_array_argument (unsigned &idx, const ICLArray< T > *array, const Strides &strides, unsigned int num_dimensions, const Window &window) |
| Add the passed array's parameters to the object's kernel's arguments starting from the index idx. More...
|
|
| IKernel () |
| Constructor. More...
|
|
virtual | ~IKernel ()=default |
| Destructor. More...
|
|
virtual bool | is_parallelisable () const |
| Indicates whether or not the kernel is parallelisable. More...
|
|
virtual BorderSize | border_size () const |
| The size of the border for that kernel. More...
|
|
const Window & | window () const |
| The maximum window the kernel can be executed on. More...
|
|
Interface for the depthwise weights reshape kernel.
This kernel reshape original weights' low 2D dimensions into a single row and have the second dimension as the original depth size.
Definition at line 38 of file CLDepthwiseWeightsReshapeKernel.h.