24.02.1
|
Basic function to run kernels::CpuCastKernel. More...
#include <CpuCast.h>
Public Member Functions | |
void | configure (const ITensorInfo *src, ITensorInfo *dst, ConvertPolicy policy) |
Configure operator for a given list of arguments. More... | |
Public Member Functions inherited from INEOperator | |
INEOperator (IRuntimeContext *ctx=nullptr) | |
Constructor. More... | |
INEOperator (const INEOperator &)=delete | |
Prevent instances of this class from being copied (As this class contains pointers) More... | |
INEOperator (INEOperator &&)=default | |
Default move constructor. More... | |
INEOperator & | operator= (const INEOperator &)=delete |
Prevent instances of this class from being copied (As this class contains pointers) More... | |
INEOperator & | operator= (INEOperator &&)=default |
Default move assignment operator. More... | |
~INEOperator () | |
Default destructor. More... | |
void | run (ITensorPack &tensors) override |
Run the kernels contained in the function. 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... | |
Public Member Functions inherited from IOperator | |
virtual | ~IOperator ()=default |
Destructor. More... | |
Static Public Member Functions | |
static Status | validate (const ITensorInfo *src, const ITensorInfo *dst, ConvertPolicy policy) |
Static function to check if given info will lead to a valid configuration. More... | |
Basic function to run kernels::CpuCastKernel.
void configure | ( | const ITensorInfo * | src, |
ITensorInfo * | dst, | ||
ConvertPolicy | policy | ||
) |
Configure operator for a given list of arguments.
Input data type must be different than output data type.
Valid data layouts:
Valid data type configurations:
src | dst |
---|---|
QASYMM8_SIGNED | S16, S32, F32, F16 |
QASYMM8 | U16, S16, S32, F32, F16 |
U8 | U16, S16, S32, F32, F16 |
U16 | U8, U32 |
S16 | QASYMM8_SIGNED, U8, S32 |
F16 | QASYMM8_SIGNED, QASYMM8, F32, S32, U8 |
S32 | QASYMM8_SIGNED, QASYMM8, F16, F32, U8 |
F32 | QASYMM8_SIGNED, QASYMM8, F16, S32, U8 |
S64 | F32 |
[in] | src | The source tensor to convert. Data types supported: U8/S8/U16/S16/U32/S32/S64/F16/F32. |
[out] | dst | The destination tensor. Data types supported: U8/S8/U16/S16/U32/S32/F16/F32. |
[in] | policy | Conversion policy. |
Definition at line 33 of file CpuCast.cpp.
References ARM_COMPUTE_LOG_PARAMS, arm_compute::test::validation::dst, and arm_compute::test::validation::src.
|
static |
Static function to check if given info will lead to a valid configuration.
Similar to CpuCast::configure()
Definition at line 41 of file CpuCast.cpp.
References arm_compute::test::validation::dst, arm_compute::test::validation::src, and CpuCastKernel::validate().
Referenced by NECast::validate(), and NEDepthConvertLayer::validate().