Compute Library
 21.08
EltwiseLayerNode.cpp
Go to the documentation of this file.
1 /*
2  * Copyright (c) 2018-2020 Arm Limited.
3  *
4  * SPDX-License-Identifier: MIT
5  *
6  * Permission is hereby granted, free of charge, to any person obtaining a copy
7  * of this software and associated documentation files (the "Software"), to
8  * deal in the Software without restriction, including without limitation the
9  * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
10  * sell copies of the Software, and to permit persons to whom the Software is
11  * furnished to do so, subject to the following conditions:
12  *
13  * The above copyright notice and this permission notice shall be included in all
14  * copies or substantial portions of the Software.
15  *
16  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
22  * SOFTWARE.
23  */
25 
29 
30 namespace arm_compute
31 {
32 namespace graph
33 {
35  : descriptor(descriptor)
36 {
37  _input_edges.resize(2, EmptyEdgeID);
38  _outputs.resize(1, NullTensorID);
39 }
40 
42 {
43  return descriptor.op;
44 }
45 
47 {
48  return descriptor.c_policy;
49 }
50 
52 {
53  return descriptor.r_policy;
54 }
55 
57 {
58  return descriptor.fused_activation;
59 }
60 
62 {
63  return descriptor.out_quant_info;
64 }
65 
67 {
69 }
70 
72 {
73  if((input_id(0) != NullTensorID) && (input_id(1) != NullTensorID) && (output_id(0) != NullTensorID))
74  {
75  Tensor *dst = output(0);
76  ARM_COMPUTE_ERROR_ON(dst == nullptr);
77  dst->desc() = configure_output(0);
78  return true;
79  }
80  return false;
81 }
82 
84 {
85  ARM_COMPUTE_UNUSED(idx);
86 
87  const Tensor *src1 = input(0);
88  ARM_COMPUTE_ERROR_ON(src1 == nullptr);
89 
90  const Tensor *src2 = input(1);
91  ARM_COMPUTE_ERROR_ON(src2 == nullptr);
92 
93  auto output_info = src1->desc();
94 
95  TensorShape out_shape = TensorShape::broadcast_shape(src1->desc().shape, src2->desc().shape);
96  ARM_COMPUTE_ERROR_ON_MSG(out_shape.total_size() == 0, "Inputs are not broadcast compatible");
97 
98  output_info.set_shape(out_shape);
99 
100  if(!descriptor.out_quant_info.empty())
101  {
102  output_info.set_quantization_info(descriptor.out_quant_info);
103  }
104 
105  return output_info;
106 }
107 
109 {
110  return NodeType::EltwiseLayer;
111 }
112 
114 {
115  v.visit(*this);
116 }
117 
119  : descriptor(descriptor)
120 {
121  _input_edges.resize(1, EmptyEdgeID);
122  _outputs.resize(1, NullTensorID);
123 }
124 
126 {
127  return descriptor;
128 }
129 
131 {
132  descriptor.fused_activation = fused_activation;
133 }
134 
136 {
137  if((input_id(0) != NullTensorID) && (output_id(0) != NullTensorID))
138  {
139  Tensor *dst = output(0);
140  ARM_COMPUTE_ERROR_ON(dst == nullptr);
141  dst->desc() = configure_output(0);
142  return true;
143  }
144  return false;
145 }
146 
148 {
149  ARM_COMPUTE_UNUSED(idx);
150 
151  const Tensor *src = input(0);
152  ARM_COMPUTE_ERROR_ON(src == nullptr);
153 
154  auto output_info = src->desc();
155 
156  if(!descriptor.out_quant_info.empty())
157  {
158  output_info.set_quantization_info(descriptor.out_quant_info);
159  }
160 
161  return output_info;
162 }
163 
165 {
167 }
168 
170 {
171  v.visit(*this);
172 }
173 
174 } // namespace graph
175 } // namespace arm_compute
TensorDescriptor configure_output(size_t idx) const override
Calculates output configuration.
NodeType type() const override
Returns node's type.
void set_fused_activation(ActivationLayerInfo fused_activation)
Sets fused activation.
Shape of a tensor.
Definition: TensorShape.h:39
EltwiseOperation
Supported Element-wise operations.
Definition: Types.h:108
RoundingPolicy rounding_policy() const
Rounding policy accessor.
static TensorShape broadcast_shape(const Shapes &... shapes)
If shapes are broadcast compatible, return the broadcasted shape.
Definition: TensorShape.h:211
#define ARM_COMPUTE_ERROR_ON(cond)
If the condition is true then an error message is printed and an exception thrown.
Definition: Error.h:466
ActivationLayerInfo fused_activation
Fused activation info.
EltwiseOperation eltwise_operation() const
Eltwise operation accessor.
Activation Layer Information class.
Definition: Types.h:1475
SimpleTensor< float > src
Definition: DFT.cpp:155
Copyright (c) 2017-2021 Arm Limited.
descriptors::UnaryEltwiseLayerDescriptor eltwise_descriptor() const
Unary eltwise layer descriptor.
bool forward_descriptors() override
Forwards descriptor information to outputs if possible.
QuantizationInfo out_quant_info
Output quantization information.
TensorDescriptor & desc()
TensorInfo metadata accessor.
Definition: Tensor.cpp:40
Quantization information.
QuantizationInfo output_quant_info() const
Returns output quantization info.
#define ARM_COMPUTE_UNUSED(...)
To avoid unused variables warnings.
Definition: Error.h:152
virtual void visit(INode &n)=0
Visit INode.
Tensor * output(size_t idx) const
Returns the tensor of a given output of the node.
Definition: INode.cpp:158
ActivationLayerInfo fused_activation() const
Returns fused activation.
ActivationLayerInfo fused_activation
Fused activation info.
#define ARM_COMPUTE_ERROR_ON_MSG(cond, msg)
Definition: Error.h:456
NodeType type() const override
Returns node&#39;s type.
size_t total_size() const
Collapses all dimensions to a single linear total size.
Definition: TensorShape.h:172
RoundingPolicy
Rounding method.
Definition: Rounding.h:30
TensorID input_id(size_t idx) const
Returns the tensor ID of a given input of the node.
Definition: INode.cpp:137
NodeType
Supported nodes.
Definition: Types.h:149
constexpr EdgeID EmptyEdgeID
Definition: Types.h:77
bool forward_descriptors() override
Forwards descriptor information to outputs if possible.
bool empty() const
Indicates whether this QuantizationInfo has valid settings or not.
void set_fused_activation(ActivationLayerInfo fused_activation)
Sets fused activation.
TensorID output_id(size_t idx) const
Returns the tensor ID of a given output of the node.
Definition: INode.cpp:144
ConvertPolicy convert_policy() const
Convert policy accessor.
QuantizationInfo out_quant_info
Output quantization information.
constexpr TensorID NullTensorID
Constant NodeID specifying an equivalent of null node.
Definition: Types.h:73
TensorDescriptor configure_output(size_t idx) const override
Calculates output configuration.
Node visitor interface.
Definition: INodeVisitor.h:34
Tensor * input(size_t idx) const
Returns the tensor of a given input of the node.
Definition: INode.cpp:150
EltwiseOperation op
Element-wise operation to perform.
ConvertPolicy
Policy to handle overflow.
Definition: Types.h:382
UnaryEltwiseLayerNode(const descriptors::UnaryEltwiseLayerDescriptor &descriptor)
Constructor.
void accept(INodeVisitor &v) override
Accepts a node visitor.
EltwiseLayerNode(const descriptors::EltwiseLayerDescriptor &descriptor)
Constructor.
Tensor object.
Definition: Tensor.h:41
void accept(INodeVisitor &v) override
Accepts a node visitor.