23.05
|
Basic function to simulate a scale layer. More...
#include <ClScale.h>
Public Member Functions | |
ClScale ()=default | |
Constructor. More... | |
void | configure (const CLCompileContext &compile_context, ITensorInfo *src, ITensorInfo *dst, const ScaleKernelInfo &info) |
Initialize the function's source, destination, interpolation type and border_mode. More... | |
void | run (ITensorPack &tensors) override |
Run the kernels contained in the function. More... | |
![]() | |
ICLOperator (IRuntimeContext *ctx=nullptr) | |
Constructor. More... | |
ICLOperator (const ICLOperator &)=delete | |
Prevent instances of this class from being copied (As this class contains pointers) More... | |
ICLOperator (ICLOperator &&)=default | |
Default move constructor. More... | |
ICLOperator & | operator= (const ICLOperator &)=delete |
Prevent instances of this class from being copied (As this class contains pointers) More... | |
ICLOperator & | operator= (ICLOperator &&)=default |
Default move assignment operator. More... | |
void | prepare (ITensorPack &constants) override |
Prepare the function for executing. More... | |
MemoryRequirements | workspace () const override |
Return the memory requirements required by the workspace. More... | |
![]() | |
virtual | ~IOperator ()=default |
Destructor. More... | |
Static Public Member Functions | |
static Status | validate (const ITensorInfo *src, const ITensorInfo *dst, const ScaleKernelInfo &info) |
Static function to check if given info will lead to a valid configuration. More... | |
Basic function to simulate a scale layer.
This function calls the following OpenCL kernels:
|
default |
Constructor.
void configure | ( | const CLCompileContext & | compile_context, |
ITensorInfo * | src, | ||
ITensorInfo * | dst, | ||
const ScaleKernelInfo & | info | ||
) |
Initialize the function's source, destination, interpolation type and border_mode.
[in] | compile_context | The compile context to be used. |
[in,out] | src | Source tensor info. Data types supported: U8/QASYMM8/QASYMM8_SIGNED/S16/F16/F32. (Written to only for border_mode != UNDEFINED) |
[out] | dst | Destination tensor info. Data types supported: Same as src All but the lowest two dimensions must be the same size as in the input tensor, i.e. scaling is only performed within the XY-plane. |
[in] | info | ScaleKernelInfo descriptor to be used to configure |
Definition at line 38 of file ClScale.cpp.
References ARM_COMPUTE_ERROR_ON_NULLPTR, ARM_COMPUTE_LOG_PARAMS, CLScheduler::get(), arm_compute::test::validation::k, and CLScheduler::tune_kernel_static().
|
overridevirtual |
Run the kernels contained in the function.
[in] | tensors | Vector that contains the tensors to operate on. |
Reimplemented from ICLOperator.
Definition at line 58 of file ClScale.cpp.
References ARM_COMPUTE_ERROR_ON_MSG, ITensorPack::empty(), CLScheduler::enqueue_op(), and CLScheduler::get().
|
static |
Static function to check if given info will lead to a valid configuration.
Similar to ClScale::configure()
Definition at line 53 of file ClScale.cpp.
References ClScaleKernel::validate().
Referenced by CLScale::validate().