24 #ifndef ARM_COMPUTE_CPU_SOFTMAX_H 25 #define ARM_COMPUTE_CPU_SOFTMAX_H 38 class CpuLogits1DMaxKernel;
39 template <
bool IS_LOG>
56 template <
bool IS_LOG = false>
92 std::unique_ptr<ICpuKernel> _max_kernel;
93 std::unique_ptr<ICpuKernel> _softmax_kernel;
94 std::unique_ptr<ITensorInfo> _max;
95 std::unique_ptr<ITensorInfo> _tmp;
96 std::unique_ptr<ITensorInfo> _input_permuted;
97 std::unique_ptr<ITensorInfo> _output_permuted;
Basic interface for functions which have a single async Neon kernel.
static Status validate(const ITensorInfo *src, const ITensorInfo *dst, float beta=1.0f, int32_t axis=0)
Static function to check if given info will lead to a valid configuration of CpuSoftmax.
Store the tensor's metadata.
SimpleTensor< float > src
Copyright (c) 2017-2021 Arm Limited.
std::vector< MemoryInfo > MemoryRequirements
void configure(const ITensorInfo *src, ITensorInfo *dst, float beta=1.0f, int32_t axis=0)
Set the input and output tensors.
CpuSoftmaxGeneric()
Constructor.
Basic function to run kernels::CpuPermuteKernel.
void run(ITensorPack &tensors) override
Run the kernels contained in the function.
experimental::MemoryRequirements workspace() const override
Return the memory requirements required by the workspace.
Basic function to compute a SoftmaxLayer and a Log SoftmaxLayer.