Compute Library
ArgMinMaxLayer Class Referencefinal

ArgMinMax Layer. More...

#include <Layers.h>

Collaboration diagram for ArgMinMaxLayer:

Public Member Functions

 ArgMinMaxLayer (ReductionOperation op, unsigned int axis, DataType out_data_type=DataType::UNKNOWN, const QuantizationInfo out_quant_info=QuantizationInfo())
 Construct an activation layer. More...
NodeID create_layer (IStream &s) override
 Create layer and add to the given stream. More...
- Public Member Functions inherited from ILayer
virtual ~ILayer ()=default
 Default destructor. More...
ILayerset_name (std::string name)
 Sets the name of the layer. More...
const std::string & name () const
 Layer name accessor. More...

Detailed Description

ArgMinMax Layer.

Definition at line 149 of file Layers.h.

Constructor & Destructor Documentation

◆ ArgMinMaxLayer()

ArgMinMaxLayer ( ReductionOperation  op,
unsigned int  axis,
DataType  out_data_type = DataType::UNKNOWN,
const QuantizationInfo  out_quant_info = QuantizationInfo() 

Construct an activation layer.

[in]opReduction Operation: min or max
[in]axisAxis to perform reduction along
[in]out_data_type(Optional) Output tensor data type
[in]out_quant_info(Optional) Output quantization info

Definition at line 159 of file Layers.h.

163  : _op(op),
164  _axis(axis),
165  _out_data_type(out_data_type),
166  _out_quant_info(std::move(out_quant_info))
167  {
168  }

Member Function Documentation

◆ create_layer()

NodeID create_layer ( IStream s)

Create layer and add to the given stream.

[in]sStream to add layer to.
ID of the created node.

Implements ILayer.

Definition at line 176 of file Layers.h.

References GraphBuilder::add_arg_min_max_node(), IStream::graph(), IStream::hints(), arm_compute::test::validation::input, ILayer::name(), IStream::tail_node(), and StreamHints::target_hint.

177  {
178  NodeParams common_params = { name(), s.hints().target_hint };
179  NodeIdxPair input = { s.tail_node(), 0 };
180  return GraphBuilder::add_arg_min_max_node(s.graph(), common_params, input, _op, _axis, _out_data_type, std::move(_out_quant_info));
181  }
const std::string & name() const
Layer name accessor.
Definition: ILayer.h:64
static NodeID add_arg_min_max_node(Graph &g, NodeParams params, NodeIdxPair input, ReductionOperation op, unsigned int axis, DataType out_data_type=DataType::UNKNOWN, const QuantizationInfo &out_quant_info=QuantizationInfo())
Adds an activation layer node to the graph.

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