24.02.1
|
Go to the documentation of this file.
38 : _op(op), _axis(axis), _out_data_type(out_data_type), _out_quant_info(std::move(out_quant_info))
56 return _out_data_type;
80 if (!_out_quant_info.
empty())
constexpr EdgeID EmptyEdgeID
SimpleTensor< float > src
Quantization information.
TensorID input_id(size_t idx) const
Returns the tensor ID of a given input of the node.
Tensor * output(size_t idx) const
Returns the tensor of a given output of the node.
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
#define ARM_COMPUTE_ERROR_ON(cond)
If the condition is true then an error message is printed and an exception thrown.
DataType out_data_type() const
Output data type accessor.
ReductionOperation reduction_operation() const
Operator accessor.
virtual void visit(INode &n)=0
Visit INode.
TensorShape compute_reduced_shape(const TensorShape &input, unsigned int axis, bool keep_dims=true)
Calculate the reduced shape of a tensor given an axis.
#define ARM_COMPUTE_UNUSED(...)
To avoid unused variables warnings.
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 empty() const
Indicates whether this QuantizationInfo has valid settings or not.
Tensor * input(size_t idx) const
Returns the tensor of a given input of the node.
TensorID output_id(size_t idx) const
Returns the tensor ID of a given output of the node.
constexpr TensorID NullTensorID
Constant NodeID specifying an equivalent of null node.
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.