24.02.1
|
#include <ClSoftmax.h>
Public Member Functions | |
ClSoftmax () | |
Constructor. More... | |
void | configure (const CLCompileContext &compile_context, const ITensorInfo &src, ITensorInfo &dst, const SoftmaxKernelInfo &info) |
Configure the operator. More... | |
void | run (ITensorPack &tensors) override |
Run the kernels contained in the function. More... | |
experimental::MemoryRequirements | workspace () const override |
Return the memory requirements required by the workspace. More... | |
Public Member Functions inherited from ICLOperator | |
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... | |
Public Member Functions inherited from IOperator | |
virtual | ~IOperator ()=default |
Destructor. More... | |
Static Public Member Functions | |
static Status | validate (const ITensorInfo &src, const ITensorInfo &dst, const SoftmaxKernelInfo &info) |
Static function to check if given info will lead to a valid configuration. More... | |
Definition at line 45 of file ClSoftmax.h.
ClSoftmax | ( | ) |
Constructor.
Definition at line 42 of file ClSoftmax.cpp.
void configure | ( | const CLCompileContext & | compile_context, |
const ITensorInfo & | src, | ||
ITensorInfo & | dst, | ||
const SoftmaxKernelInfo & | info | ||
) |
Configure the operator.
[in] | compile_context | The compile context to be used. |
[in] | src | Source tensor info. Data types supported: QASYMM8/QASYMM8_SIGNED/F16/F32 for Softmax and F16/F32 for Log Softmax |
[out] | dst | Destination tensor info. Data types supported: same as src |
[in] | info | Contains information consumed by kernels for softmax described in SoftmaxKernelInfo. |
Definition at line 46 of file ClSoftmax.cpp.
References ARM_COMPUTE_LOG_PARAMS, arm_compute::test::validation::dst, arm_compute::test::validation::info, arm_compute::offset_int_vec(), arm_compute::test::validation::src, and TensorInfo::total_size().
|
overridevirtual |
Run the kernels contained in the function.
[in] | tensors | Vector that contains the tensors to operate on. |
Reimplemented from ICLOperator.
Definition at line 69 of file ClSoftmax.cpp.
References arm_compute::ACL_INT_0, ITensorPack::add_tensor(), CLScheduler::enqueue_op(), CLScheduler::get(), CLAuxTensorHandler::get(), and arm_compute::offset_int_vec().
|
static |
Static function to check if given info will lead to a valid configuration.
Similar to ClSoftmax::configure()
Definition at line 64 of file ClSoftmax.cpp.
References arm_compute::test::validation::dst, arm_compute::test::validation::info, arm_compute::test::validation::src, and ClSoftmaxKernel::validate().
|
overridevirtual |
Return the memory requirements required by the workspace.
Reimplemented from ICLOperator.
Definition at line 78 of file ClSoftmax.cpp.