24.02.1
|
Go to the documentation of this file.
24 #ifndef ARM_COMPUTE_GRAPH_ARGMINMAX_LAYER_NODE_H
25 #define ARM_COMPUTE_GRAPH_ARGMINMAX_LAYER_NODE_H
57 unsigned int axis()
const;
Quantization information.
TensorDescriptor configure_output(size_t idx) const override
Calculates output configuration.
NodeType type() const override
Returns node's type.
ReductionOperation
Available reduction operations.
static constexpr NodeType node_type
DataType out_data_type() const
Output data type accessor.
ReductionOperation reduction_operation() const
Operator accessor.
ArgMinMaxLayerNode(ReductionOperation op, unsigned int axis, DataType out_data_type=DataType::UNKNOWN, QuantizationInfo out_quant_info=QuantizationInfo())
Constructor.
unsigned int axis() const
Axis accessor.
Copyright (c) 2017-2024 Arm Limited.
bool forward_descriptors() override
Forwards descriptor information to outputs if possible.
void accept(INodeVisitor &v) override
Accepts a node visitor.
@ UNKNOWN
Unknown data type.
DataType
Available data types.