diff options
Diffstat (limited to 'arm_compute')
-rw-r--r-- | arm_compute/core/GLES_COMPUTE/GCKernels.h | 2 | ||||
-rw-r--r-- | arm_compute/core/GLES_COMPUTE/kernels/GCDropoutLayerKernel.h (renamed from arm_compute/core/GLES_COMPUTE/kernels/GCDropoutKernel.h) | 20 | ||||
-rw-r--r-- | arm_compute/runtime/GLES_COMPUTE/functions/GCDropoutLayer.h | 6 |
3 files changed, 14 insertions, 14 deletions
diff --git a/arm_compute/core/GLES_COMPUTE/GCKernels.h b/arm_compute/core/GLES_COMPUTE/GCKernels.h index 9831e25299..417c98af67 100644 --- a/arm_compute/core/GLES_COMPUTE/GCKernels.h +++ b/arm_compute/core/GLES_COMPUTE/GCKernels.h @@ -31,7 +31,7 @@ #include "arm_compute/core/GLES_COMPUTE/kernels/GCCol2ImKernel.h" #include "arm_compute/core/GLES_COMPUTE/kernels/GCDepthConcatenateLayerKernel.h" #include "arm_compute/core/GLES_COMPUTE/kernels/GCDirectConvolutionLayerKernel.h" -#include "arm_compute/core/GLES_COMPUTE/kernels/GCDropoutKernel.h" +#include "arm_compute/core/GLES_COMPUTE/kernels/GCDropoutLayerKernel.h" #include "arm_compute/core/GLES_COMPUTE/kernels/GCFillBorderKernel.h" #include "arm_compute/core/GLES_COMPUTE/kernels/GCGEMMInterleave4x4Kernel.h" #include "arm_compute/core/GLES_COMPUTE/kernels/GCGEMMMatrixAccumulateBiasesKernel.h" diff --git a/arm_compute/core/GLES_COMPUTE/kernels/GCDropoutKernel.h b/arm_compute/core/GLES_COMPUTE/kernels/GCDropoutLayerKernel.h index 6159a7af26..9f04411d92 100644 --- a/arm_compute/core/GLES_COMPUTE/kernels/GCDropoutKernel.h +++ b/arm_compute/core/GLES_COMPUTE/kernels/GCDropoutLayerKernel.h @@ -22,8 +22,8 @@ * SOFTWARE. */ -#ifndef __ARM_COMPUTE_GCDROPOUTKERNEL_H__ -#define __ARM_COMPUTE_GCDROPOUTKERNEL_H__ +#ifndef __ARM_COMPUTE_GCDROPOUTLAYERKERNEL_H__ +#define __ARM_COMPUTE_GCDROPOUTLAYERKERNEL_H__ #include "arm_compute/core/GLES_COMPUTE/IGCKernel.h" @@ -31,28 +31,28 @@ namespace arm_compute { class IGCTensor; -/** Interface for the dropout kernel. +/** Interface for the dropout layer kernel. * * Dropout is used to improve over-fit on neural networks. * */ -class GCDropoutKernel : public IGCKernel +class GCDropoutLayerKernel : public IGCKernel { public: /** Default constructor */ - GCDropoutKernel(); + GCDropoutLayerKernel(); /** Prevent instances of this class from being copied (As this class contains pointers) */ - GCDropoutKernel(const GCDropoutKernel &) = delete; + GCDropoutLayerKernel(const GCDropoutLayerKernel &) = delete; /** Prevent instances of this class from being copied (As this class contains pointers) */ - GCDropoutKernel &operator=(const GCDropoutKernel &) = delete; + GCDropoutLayerKernel &operator=(const GCDropoutLayerKernel &) = delete; /** Allow instances of this class to be moved */ - GCDropoutKernel(GCDropoutKernel &&) = default; + GCDropoutLayerKernel(GCDropoutLayerKernel &&) = default; /** Allow instances of this class to be moved */ - GCDropoutKernel &operator=(GCDropoutKernel &&) = default; + GCDropoutLayerKernel &operator=(GCDropoutLayerKernel &&) = default; /** Set the input and output of the kernel. * @@ -76,4 +76,4 @@ private: }; } -#endif /*__ARM_COMPUTE_GCDROPOUTKERNEL_H__ */ +#endif /*__ARM_COMPUTE_GCDROPOUTLAYERKERNEL_H__ */ diff --git a/arm_compute/runtime/GLES_COMPUTE/functions/GCDropoutLayer.h b/arm_compute/runtime/GLES_COMPUTE/functions/GCDropoutLayer.h index 6a08d96676..c51d2c1613 100644 --- a/arm_compute/runtime/GLES_COMPUTE/functions/GCDropoutLayer.h +++ b/arm_compute/runtime/GLES_COMPUTE/functions/GCDropoutLayer.h @@ -25,7 +25,7 @@ #ifndef __ARM_COMPUTE_GCDROPOUTLAYER_H__ #define __ARM_COMPUTE_GCDROPOUTLAYER_H__ -#include "arm_compute/core/GLES_COMPUTE/kernels/GCDropoutKernel.h" +#include "arm_compute/core/GLES_COMPUTE/kernels/GCDropoutLayerKernel.h" #include "arm_compute/runtime/IFunction.h" namespace arm_compute @@ -33,7 +33,7 @@ namespace arm_compute class IGCTensor; /** Basic function to do dropout op. This function calls the following kernels: * - * -# @ref GCDropoutKernel + * -# @ref GCDropoutLayerKernel */ class GCDropoutLayer : public IFunction { @@ -56,7 +56,7 @@ public: void run() override; private: - GCDropoutKernel _dropout_kernel; + GCDropoutLayerKernel _dropout_kernel; }; } |