24 #ifndef ARM_COMPUTE_CL_SOFTMAX_H 25 #define ARM_COMPUTE_CL_SOFTMAX_H 33 struct SoftmaxKernelInfo;
40 class ClLogits1DMaxShiftExpSumKernel;
41 class ClLogits1DNormKernel;
68 enum InternalTensorIdx
78 std::unique_ptr<ClPermute> _permute_input;
79 std::unique_ptr<ClPermute> _permute_output;
80 std::unique_ptr<kernels::ClLogits1DMaxShiftExpSumKernel> _max_shift_exp_sum_kernel;
81 std::unique_ptr<kernels::ClLogits1DNormKernel> _norm_kernel;
82 bool _needs_permute{
false };
Store the tensor's metadata.
SimpleTensor< float > src
Copyright (c) 2017-2021 Arm Limited.
std::vector< MemoryInfo > MemoryRequirements
ScaleKernelInfo info(interpolation_policy, default_border_mode, PixelValue(), sampling_policy, false)
Store the tensor's metadata.
im2col_func configure(src_target.info(), dst_target.info(), spatial_kernel, conv_info, has_bias)
Basic interface for functions which have a single async CL kernel.
Descriptor used by the softmax kernels.
Status validate(const ITensorInfo *scores_in, const ITensorInfo *boxes_in, const ITensorInfo *batch_splits_in, const ITensorInfo *scores_out, const ITensorInfo *boxes_out, const ITensorInfo *classes, const ITensorInfo *batch_splits_out, const ITensorInfo *keeps, const ITensorInfo *keeps_size, const BoxNMSLimitInfo info)