21.02
|
Basic function to do dropout op. More...
#include <GCDropoutLayer.h>
Public Member Functions | |
GCDropoutLayer () | |
Constructor. More... | |
void | configure (const IGCTensor *input, IGCTensor *mask, IGCTensor *output, float ratio, bool forward) |
Set the input and output tensors. More... | |
void | run () override |
Run the kernels contained in the function. More... | |
Public Member Functions inherited from IFunction | |
virtual | ~IFunction ()=default |
Destructor. More... | |
virtual void | prepare () |
Prepare the function for executing. More... | |
Basic function to do dropout op.
This function calls the following kernels:
Definition at line 41 of file GCDropoutLayer.h.
GCDropoutLayer | ( | ) |
void configure | ( | const IGCTensor * | input, |
IGCTensor * | mask, | ||
IGCTensor * | output, | ||
float | ratio, | ||
bool | forward | ||
) |
Set the input and output tensors.
[in] | input | Source tensor. Data type supported: F16/F32. |
[out] | mask | Destination tensor. Data type supported: Same as input . |
[out] | output | Destination tensor. Data type supported: Same as input . |
[in] | ratio | Dropout ratio |
[in] | forward | Forward or backward propagation |
Definition at line 38 of file GCDropoutLayer.cpp.
References ARM_COMPUTE_ERROR_ON_DATA_TYPE_CHANNEL_NOT_IN, ARM_COMPUTE_ERROR_ON_MISMATCHING_DATA_TYPES, GCDropoutLayerKernel::configure(), arm_compute::F16, and arm_compute::F32.
|
overridevirtual |
Run the kernels contained in the function.
For Neon kernels:
For OpenCL kernels:
Implements IFunction.
Definition at line 47 of file GCDropoutLayer.cpp.
References GCScheduler::dispatch(), and GCScheduler::get().