diff options
Diffstat (limited to 'arm_compute/core/NEON/kernels/NESoftmaxLayerKernel.h')
-rw-r--r-- | arm_compute/core/NEON/kernels/NESoftmaxLayerKernel.h | 14 |
1 files changed, 13 insertions, 1 deletions
diff --git a/arm_compute/core/NEON/kernels/NESoftmaxLayerKernel.h b/arm_compute/core/NEON/kernels/NESoftmaxLayerKernel.h index 0fecfac15a..bd0e642d76 100644 --- a/arm_compute/core/NEON/kernels/NESoftmaxLayerKernel.h +++ b/arm_compute/core/NEON/kernels/NESoftmaxLayerKernel.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017 ARM Limited. + * Copyright (c) 2017-2018 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -35,6 +35,10 @@ class ITensor; class NELogits1DMaxKernel : public INESimpleKernel { public: + const char *name() const override + { + return "NELogits1DMaxKernel"; + } /** Default constructor */ NELogits1DMaxKernel(); /** Set the input and output tensors. @@ -68,6 +72,10 @@ private: class NELogits1DShiftExpSumKernel : public INEKernel { public: + const char *name() const override + { + return "NELogits1DShiftExpSumKernel"; + } /** Default constructor */ NELogits1DShiftExpSumKernel(); /** Prevent instances of this class from being copied (As this class contains pointers) */ @@ -120,6 +128,10 @@ private: class NELogits1DNormKernel : public INEKernel { public: + const char *name() const override + { + return "NELogits1DNormKernel"; + } /** Default constructor */ NELogits1DNormKernel(); /** Prevent instances of this class from being copied (As this class contains pointers) */ |