diff options
Diffstat (limited to 'arm_compute/runtime/GLES_COMPUTE')
-rw-r--r-- | arm_compute/runtime/GLES_COMPUTE/functions/GCDropoutLayer.h | 6 |
1 files changed, 3 insertions, 3 deletions
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; }; } |