ONE - On-device Neural Engine
|
#include <CLScaleFactorSymm8Kernel.h>
Public Member Functions | |
CLScaleFactorSymm8Kernel () | |
CLScaleFactorSymm8Kernel (const CLScaleFactorSymm8Kernel &)=delete | |
CLScaleFactorSymm8Kernel & | operator= (const CLScaleFactorSymm8Kernel &)=delete |
CLScaleFactorSymm8Kernel (CLScaleFactorSymm8Kernel &&)=default | |
CLScaleFactorSymm8Kernel & | operator= (CLScaleFactorSymm8Kernel &&)=default |
void | configure (const ICLTensor *input, ICLTensor *output) |
void | reset (cl::CommandQueue &queue) |
void | run (const Window &window, cl::CommandQueue &queue) override |
Static Public Member Functions | |
static Status | validate (const ITensorInfo *input, const ITensorInfo *output) |
Interface for the kernel to perform min max search on a 3D tensor.
Definition at line 52 of file CLScaleFactorSymm8Kernel.h.
CLScaleFactorSymm8Kernel::CLScaleFactorSymm8Kernel | ( | ) |
Default constructor
Definition at line 107 of file CLScaleFactorSymm8Kernel.cpp.
|
delete |
Prevent instances of this class from being copied (As this class contains pointers)
|
default |
Allow instances of this class to be moved
void CLScaleFactorSymm8Kernel::configure | ( | const ICLTensor * | input, |
ICLTensor * | output | ||
) |
Initialise the kernel's input and output.
[in] | input | Input tensor with 2 dimensions. The first dimension will be interpreted as batches. Data types supported: F32. |
[out] | output | Output tensor with shape [batches] which stores the scale values for each 2D input tensor. The dimensions over the first must match the batched dimensions of the input tensor. Data types supported: F32. |
Definition at line 109 of file CLScaleFactorSymm8Kernel.cpp.
References arm_compute::CLKernelLibraryEx::create_kernel(), and arm_compute::CLKernelLibraryEx::get().
Referenced by arm_compute::CLFullyConnectedHybridLayer::configure().
|
default |
Allow instances of this class to be moved
References validate().
|
delete |
Prevent instances of this class from being copied (As this class contains pointers)
void arm_compute::CLScaleFactorSymm8Kernel::reset | ( | cl::CommandQueue & | queue | ) |
Resets global minimum and maximum
[in,out] | queue | Command queue on which to map and unmap the min_max tensor |
|
override |
Definition at line 140 of file CLScaleFactorSymm8Kernel.cpp.
Referenced by package.infer.session::inference().
|
static |
Static function to check if given info will lead to a valid configuration of CLScaleFactorSymm8Kernel
[in] | input | Input tensor info. Data types supported: F32. |
[in] | output | Output tensor info with shape [batches] which stores the scale values for each 2D input tensor. The dimensions over the first must match the batched dimensions of the input tensor. Data types supported: F32. |
Definition at line 131 of file CLScaleFactorSymm8Kernel.cpp.
Referenced by operator=(), and arm_compute::CLFullyConnectedHybridLayer::validate().