36 : _op(op), _axis(axis), _out_data_type(out_data_type), _out_quant_info(
std::move(out_quant_info))
54 return _out_data_type;
78 if(!_out_quant_info.
empty())
DataType data_type
Data type.
TensorShape shape
Tensor shape.
DataType out_data_type() const
Output data type accessor.
ReductionOperation
Available reduction operations.
ArgMinMaxLayerNode(ReductionOperation op, unsigned int axis, DataType out_data_type=DataType::UNKNOWN, QuantizationInfo out_quant_info=QuantizationInfo())
Constructor.
#define ARM_COMPUTE_ERROR_ON(cond)
If the condition is true then an error message is printed and an exception thrown.
SimpleTensor< float > src
Copyright (c) 2017-2021 Arm Limited.
bool forward_descriptors() override
Forwards descriptor information to outputs if possible.
void accept(INodeVisitor &v) override
Accepts a node visitor.
TensorDescriptor & desc()
TensorInfo metadata accessor.
Quantization information.
QuantizationInfo quant_info
Quantization info.
#define ARM_COMPUTE_UNUSED(...)
To avoid unused variables warnings.
virtual void visit(INode &n)=0
Visit INode.
Tensor * output(size_t idx) const
Returns the tensor of a given output of the node.
NodeType type() const override
Returns node's type.
TensorShape compute_reduced_shape(const TensorShape &input, unsigned int axis, bool keep_dims=true)
Calculate the reduced shape of a tensor given an axis.
TensorID input_id(size_t idx) const
Returns the tensor ID of a given input of the node.
constexpr EdgeID EmptyEdgeID
TensorDescriptor configure_output(size_t idx) const override
Calculates output configuration.
bool empty() const
Indicates whether this QuantizationInfo has valid settings or not.
TensorDescriptor & set_shape(TensorShape &tensor_shape)
Sets tensor descriptor shape.
TensorID output_id(size_t idx) const
Returns the tensor ID of a given output of the node.
ReductionOperation reduction_operation() const
Operator accessor.
unsigned int axis() const
Axis accessor.
constexpr TensorID NullTensorID
Constant NodeID specifying an equivalent of null node.
Tensor * input(size_t idx) const
Returns the tensor of a given input of the node.
DataType
Available data types.
static constexpr NodeType node_type