24.02.1
|
Go to the documentation of this file.
24 #ifndef ACL_SRC_CPU_OPERATORS_CPUSOFTMAX_H
25 #define ACL_SRC_CPU_OPERATORS_CPUSOFTMAX_H
40 class CpuSoftmaxKernel;
84 enum InternalTensorIdx
94 std::unique_ptr<ICPPKernel> _softmax_kernel;
106 #endif // ACL_SRC_CPU_OPERATORS_CPUSOFTMAX_H
std::vector< MemoryInfo > MemoryRequirements
SimpleTensor< float > src
experimental::MemoryRequirements workspace() const override
Return the memory requirements required by the workspace.
void run(ITensorPack &tensors) override
Run the kernels contained in the function.
void configure(const ITensorInfo *src, ITensorInfo *dst, float beta=1.0f, int32_t axis=0, bool is_log=false)
Set the input and output tensors.
Basic function to compute a SoftmaxLayer and a Log SoftmaxLayer.
Basic interface for functions which have a single async CPU kernel.
Store the tensor's metadata.
Basic function to run kernels::CpuPermuteKernel.
Copyright (c) 2017-2024 Arm Limited.
static Status validate(const ITensorInfo *src, const ITensorInfo *dst, float beta=1.0f, int32_t axis=0, bool is_log=false)
Static function to check if given info will lead to a valid configuration.
Store the tensor's metadata.