diff options
author | Georgios Pinitas <georgios.pinitas@arm.com> | 2021-07-02 09:01:49 +0100 |
---|---|---|
committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2021-07-02 15:47:11 +0000 |
commit | 2eb5d16b839cbc28c6cb7f0de7a0bf15290b425a (patch) | |
tree | 523d495c3a4c07d87b337c45a81afa06c9b1f495 /src/runtime/cpu/operators/CpuSoftmax.h | |
parent | 4dfc5538948c196def6d2e3305fe8051a5df3f15 (diff) | |
download | ComputeLibrary-2eb5d16b839cbc28c6cb7f0de7a0bf15290b425a.tar.gz |
Align kernel/operator header layout
- Redirect validate documentation to configure
- Align header names
- Align class layout
Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com>
Change-Id: Ia40f67383826a66e9f9a33745d66805551e31a3a
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5897
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Diffstat (limited to 'src/runtime/cpu/operators/CpuSoftmax.h')
-rw-r--r-- | src/runtime/cpu/operators/CpuSoftmax.h | 10 |
1 files changed, 2 insertions, 8 deletions
diff --git a/src/runtime/cpu/operators/CpuSoftmax.h b/src/runtime/cpu/operators/CpuSoftmax.h index 38817977b3..a9ac803c09 100644 --- a/src/runtime/cpu/operators/CpuSoftmax.h +++ b/src/runtime/cpu/operators/CpuSoftmax.h @@ -57,7 +57,6 @@ template <bool IS_LOG = false> class CpuSoftmaxGeneric : public ICpuOperator { public: - /** Constructor */ CpuSoftmaxGeneric(); /** Set the input and output tensors. * @@ -69,14 +68,9 @@ public: * axis=1, softmax will be applied to 4x6=24 vectors of size 5. Defaults to 0 */ void configure(const ITensorInfo *src, ITensorInfo *dst, float beta = 1.0f, int32_t axis = 0); - - /** Static function to check if given info will lead to a valid configuration of @ref CpuSoftmax + /** Static function to check if given info will lead to a valid configuration * - * @param[in] src Source tensor info. Data types supported: QASYMM8/QASYMM8_SIGNED/F16/F32. - * @param[in] dst Destination tensor info. Data types supported: same as @p input - * @param[in] beta (Optional) A scaling factor for the exponent. - * @param[in] axis (Optional) The dimension in which to apply the function. E.g. for input of shape 4x5x6 and - * axis=1, softmax will be applied to 4x6=24 vectors of size 5. Defaults to 0 + * Similar to @ref CpuSoftmaxGeneric::configure() * * @return a status */ |