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 */
24 #include "arm_compute/graph/nodes/EltwiseLayerNode.h"
25
26 #include "arm_compute/core/TensorShape.h"
27 #include "arm_compute/graph/Graph.h"
28 #include "arm_compute/graph/INodeVisitor.h"
29
30 namespace arm_compute
31 {
32 namespace graph
33 {
EltwiseLayerNode(const descriptors::EltwiseLayerDescriptor & descriptor)34 EltwiseLayerNode::EltwiseLayerNode(const descriptors::EltwiseLayerDescriptor &descriptor)
35 : descriptor(descriptor)
36 {
37 _input_edges.resize(2, EmptyEdgeID);
38 _outputs.resize(1, NullTensorID);
39 }
40
eltwise_operation() const41 EltwiseOperation EltwiseLayerNode::eltwise_operation() const
42 {
43 return descriptor.op;
44 }
45
convert_policy() const46 ConvertPolicy EltwiseLayerNode::convert_policy() const
47 {
48 return descriptor.c_policy;
49 }
50
rounding_policy() const51 RoundingPolicy EltwiseLayerNode::rounding_policy() const
52 {
53 return descriptor.r_policy;
54 }
55
fused_activation() const56 ActivationLayerInfo EltwiseLayerNode::fused_activation() const
57 {
58 return descriptor.fused_activation;
59 }
60
output_quant_info() const61 QuantizationInfo EltwiseLayerNode::output_quant_info() const
62 {
63 return descriptor.out_quant_info;
64 }
65
set_fused_activation(ActivationLayerInfo fused_activation)66 void EltwiseLayerNode::set_fused_activation(ActivationLayerInfo fused_activation)
67 {
68 descriptor.fused_activation = fused_activation;
69 }
70
forward_descriptors()71 bool EltwiseLayerNode::forward_descriptors()
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
configure_output(size_t idx) const83 TensorDescriptor EltwiseLayerNode::configure_output(size_t idx) const
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
type() const108 NodeType EltwiseLayerNode::type() const
109 {
110 return NodeType::EltwiseLayer;
111 }
112
accept(INodeVisitor & v)113 void EltwiseLayerNode::accept(INodeVisitor &v)
114 {
115 v.visit(*this);
116 }
117
UnaryEltwiseLayerNode(const descriptors::UnaryEltwiseLayerDescriptor & descriptor)118 UnaryEltwiseLayerNode::UnaryEltwiseLayerNode(const descriptors::UnaryEltwiseLayerDescriptor &descriptor)
119 : descriptor(descriptor)
120 {
121 _input_edges.resize(1, EmptyEdgeID);
122 _outputs.resize(1, NullTensorID);
123 }
124
eltwise_descriptor() const125 descriptors::UnaryEltwiseLayerDescriptor UnaryEltwiseLayerNode::eltwise_descriptor() const
126 {
127 return descriptor;
128 }
129
set_fused_activation(ActivationLayerInfo fused_activation)130 void UnaryEltwiseLayerNode::set_fused_activation(ActivationLayerInfo fused_activation)
131 {
132 descriptor.fused_activation = fused_activation;
133 }
134
forward_descriptors()135 bool UnaryEltwiseLayerNode::forward_descriptors()
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
configure_output(size_t idx) const147 TensorDescriptor UnaryEltwiseLayerNode::configure_output(size_t idx) const
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
type() const164 NodeType UnaryEltwiseLayerNode::type() const
165 {
166 return NodeType::UnaryEltwiseLayer;
167 }
168
accept(INodeVisitor & v)169 void UnaryEltwiseLayerNode::accept(INodeVisitor &v)
170 {
171 v.visit(*this);
172 }
173
174 } // namespace graph
175 } // namespace arm_compute
176