|
ONE - On-device Neural Engine
|
#include <CLMultiplyScaleFactorKernel.h>

Public Member Functions | |
| CLMultiplyScaleFactorKernel () | |
| CLMultiplyScaleFactorKernel (const CLMultiplyScaleFactorKernel &)=delete | |
| CLMultiplyScaleFactorKernel & | operator= (const CLMultiplyScaleFactorKernel &)=delete |
| CLMultiplyScaleFactorKernel (CLMultiplyScaleFactorKernel &&)=default | |
| CLMultiplyScaleFactorKernel & | operator= (CLMultiplyScaleFactorKernel &&)=default |
| ~CLMultiplyScaleFactorKernel ()=default | |
| void | configure (const ICLTensor *input, const ICLTensor *scale_factor, ICLTensor *output, float multiplier=1.f) |
| void | run (const Window &window, cl::CommandQueue &queue) override |
| Enqueue the OpenCL kernel to process the given window on the passed OpenCL command queue. | |
Static Public Member Functions | |
| static Status | validate (const ITensorInfo *input, const ITensorInfo *scale_factor, const ITensorInfo *output) |
Interface to multiply scale factor kernel.
Definition at line 51 of file CLMultiplyScaleFactorKernel.h.
| CLMultiplyScaleFactorKernel::CLMultiplyScaleFactorKernel | ( | ) |
Default constructor
Definition at line 105 of file CLMultiplyScaleFactorKernel.cpp.
|
delete |
Prevent instances of this class from being copied (As this class contains pointers)
|
default |
Default Move Constructor.
|
default |
Default destructor
References validate().
| void CLMultiplyScaleFactorKernel::configure | ( | const ICLTensor * | input, |
| const ICLTensor * | scale_factor, | ||
| ICLTensor * | output, | ||
| float | multiplier = 1.f |
||
| ) |
Set input, output tensors.
| [in/out] | input Source tensor. Data type supported: S32. | |
| [in] | scale_factor | Scale tensor. Data type supported: F16/F32. |
| [out] | output | Destination tensor. Data type supported: Same as scale_factor. |
| [in] | multiplier | Additional scale value. |
Definition at line 110 of file CLMultiplyScaleFactorKernel.cpp.
References arm_compute::CLKernelLibraryEx::create_kernel(), and arm_compute::CLKernelLibraryEx::get().
Referenced by arm_compute::CLFullyConnectedHybridLayer::configure().
|
default |
Default move assignment operator
|
delete |
Prevent instances of this class from being copied (As this class contains pointers)
|
override |
Enqueue the OpenCL kernel to process the given window on the passed OpenCL command queue.
| [in] | window | Region on which to execute the kernel. (Must be a valid region of the window returned by window()). |
| [in,out] | queue | Command queue on which to enqueue the kernel. |
Definition at line 158 of file CLMultiplyScaleFactorKernel.cpp.
|
static |
Static function to check if given info will lead to a valid configuration of CLMultiplyScaleFactorKernel
| [in] | input | Input tensor info. Data types supported: S32. |
| [in] | scale_factor | Scale tensor. Data type supported: F16/F32. |
| [in] | output | Output tensor info. Data types supported: Same as scale_factor. |
| [in] | multiplier | Additional scale value. |
Definition at line 148 of file CLMultiplyScaleFactorKernel.cpp.
Referenced by arm_compute::CLFullyConnectedHybridLayer::validate(), and ~CLMultiplyScaleFactorKernel().