Compute Library
 19.08
NEYOLOLayer Class Reference

Basic function to run NEYOLOLayerKernel. More...

#include <NEYOLOLayer.h>

Collaboration diagram for NEYOLOLayer:
[legend]

Public Member Functions

void configure (ITensor *input, ITensor *output, const ActivationLayerInfo &act_info, int32_t num_classes)
 Set the input and output tensor. More...
 
- Public Member Functions inherited from INESimpleFunctionNoBorder
 INESimpleFunctionNoBorder ()
 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...
 

Static Public Member Functions

static Status validate (const ITensorInfo *input, const ITensorInfo *output, const ActivationLayerInfo &act_info, int32_t num_classes)
 Static function to check if given info will lead to a valid configuration of NEYOLOLayer. More...
 

Detailed Description

Basic function to run NEYOLOLayerKernel.

Definition at line 37 of file NEYOLOLayer.h.

Member Function Documentation

◆ configure()

void configure ( ITensor input,
ITensor output,
const ActivationLayerInfo act_info,
int32_t  num_classes 
)

Set the input and output tensor.

Note
If the output tensor is a nullptr or is equal to the input, the activation function will be performed in-place
Parameters
[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.
[in]num_classesNumber of classes to activate (must be submultiple of input channels)

Definition at line 31 of file NEYOLOLayer.cpp.

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

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

◆ validate()

Status validate ( const ITensorInfo input,
const ITensorInfo output,
const ActivationLayerInfo act_info,
int32_t  num_classes 
)
static

Static function to check if given info will lead to a valid configuration of NEYOLOLayer.

Parameters
[in]inputSource tensor info. In case of output tensor info = nullptr, this tensor will store the result of the activation function. Data types supported: F16/F32.
[in]outputDestination tensor info. Data type supported: same as input
[in]act_infoActivation layer information.
[in]num_classesNumber of classes to activate (must be submultiple of input channels)
Returns
a status

Definition at line 38 of file NEYOLOLayer.cpp.

39 {
40  return NEYOLOLayerKernel::validate(input, output, act_info, num_classes);
41 }
static Status validate(const ITensorInfo *input, const ITensorInfo *output, const ActivationLayerInfo &act_info, int32_t num_classes)
Static function to check if given info will lead to a valid configuration of NEYOLOLayerKernel.

References arm_compute::test::validation::act_info, and NEYOLOLayerKernel::validate().


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