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.
Referenced by package.infer.session::inference().
|
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().