21.02
GCDropoutLayer.h
Go to the documentation of this file.
1
/*
2
* Copyright (c) 2017-2020 Arm Limited.
3
*
4
* SPDX-License-Identifier: MIT
5
*
6
* Permission is hereby granted, free of charge, to any person obtaining a copy
7
* of this software and associated documentation files (the "Software"), to
8
* deal in the Software without restriction, including without limitation the
9
* rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
10
* sell copies of the Software, and to permit persons to whom the Software is
11
* furnished to do so, subject to the following conditions:
12
*
13
* The above copyright notice and this permission notice shall be included in all
14
* copies or substantial portions of the Software.
15
*
16
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
22
* SOFTWARE.
23
*/
24
25
#ifndef ARM_COMPUTE_GCDROPOUTLAYER_H
26
#define ARM_COMPUTE_GCDROPOUTLAYER_H
27
28
#include "
arm_compute/core/GLES_COMPUTE/kernels/GCDropoutLayerKernel.h
"
29
#include "
arm_compute/runtime/IFunction.h
"
30
31
namespace
arm_compute
32
{
33
class
IGCTensor;
34
/** Basic function to do dropout op. This function calls the following kernels:
35
*
36
* -# @ref GCDropoutLayerKernel
37
*
38
* @deprecated This function is deprecated and is intended to be removed in 21.05 release
39
*
40
*/
41
class
GCDropoutLayer
:
public
IFunction
42
{
43
public
:
44
/** Constructor */
45
GCDropoutLayer
();
46
47
/** Set the input and output tensors.
48
*
49
* @param[in] input Source tensor. Data type supported: F16/F32.
50
* @param[out] mask Destination tensor. Data type supported: Same as @p input.
51
* @param[out] output Destination tensor. Data type supported: Same as @p input.
52
* @param[in] ratio Dropout ratio
53
* @param[in] forward Forward or backward propagation
54
*
55
*/
56
void
configure
(
const
IGCTensor
*
input
,
IGCTensor
*mask,
IGCTensor
*output,
float
ratio,
bool
forward
);
57
58
//Inherited methods override
59
void
run
()
override
;
60
61
private
:
62
GCDropoutLayerKernel
_dropout_kernel;
63
};
64
}
65
66
#endif
/* ARM_COMPUTE_GCDROPOUTLAYER_H */
arm_compute::IFunction
Base class for all functions.
Definition:
IFunction.h:30
arm_compute::IGCTensor
Interface for GLES Compute tensor.
Definition:
IGCTensor.h:35
arm_compute
Copyright (c) 2017-2021 Arm Limited.
Definition:
00_introduction.dox:24
GCDropoutLayerKernel.h
arm_compute::test::validation::input
auto input
Definition:
LSTMLayerQuantized.cpp:486
arm_compute::test::validation::forward
auto forward
Definition:
DFT.cpp:123
arm_compute::GCDropoutLayer::run
void run() override
Run the kernels contained in the function.
Definition:
GCDropoutLayer.cpp:47
arm_compute::GCDropoutLayer
Basic function to do dropout op.
Definition:
GCDropoutLayer.h:41
arm_compute::GCDropoutLayer::GCDropoutLayer
GCDropoutLayer()
Constructor.
Definition:
GCDropoutLayer.cpp:33
IFunction.h
arm_compute::GCDropoutLayerKernel
Interface for the dropout layer kernel.
Definition:
GCDropoutLayerKernel.h:39
arm_compute::GCDropoutLayer::configure
void configure(const IGCTensor *input, IGCTensor *mask, IGCTensor *output, float ratio, bool forward)
Set the input and output tensors.
Definition:
GCDropoutLayer.cpp:38
arm_compute
runtime
GLES_COMPUTE
functions
GCDropoutLayer.h
Generated on Thu Feb 25 2021 17:17:32 for Compute Library by
1.8.13