Compute Library
GCActivationLayer Class Reference

Basic function to run GCActivationLayerKernel. More...

#include <GCActivationLayer.h>

Collaboration diagram for GCActivationLayer:

Public Member Functions

void configure (IGCTensor *input, IGCTensor *output, ActivationLayerInfo act_info)
 Set the input and output tensor. More...
- Public Member Functions inherited from IGCSimpleFunction
 IGCSimpleFunction ()
 Default constructor. More...
void run () override final
 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...

Detailed Description

Basic function to run GCActivationLayerKernel.

The function simulates an activation layer with the specified activation function.

Definition at line 38 of file GCActivationLayer.h.

Member Function Documentation

◆ configure()

void configure ( IGCTensor input,
IGCTensor output,
ActivationLayerInfo  act_info 

Set the input and output tensor.

If the output tensor is a nullptr, the activation function will be performed in-place
[in,out]inputSource tensor. In case of output tensor = nullptr, this tensor will store the result of the activation function. Data types supported: F16/F32.
[out]outputDestination tensor. Data type supported: same as input
[in]act_infoActivation layer parameters.

Definition at line 32 of file GCActivationLayer.cpp.

33 {
34  auto k = arm_compute::support::cpp14::make_unique<GCActivationLayerKernel>();
35  k->configure(input, output, act_info);
36  _kernel = std::move(k);
37 }

References arm_compute::test::validation::act_info.

Referenced by GCDepthwiseConvolutionLayer3x3::configure(), and GCConvolutionLayer::configure().

The documentation for this class was generated from the following files: