From add078b7f9a658d52ee6e8fe6771ea0517c07674 Mon Sep 17 00:00:00 2001 From: mathad01 Date: Mon, 10 May 2021 16:51:20 +0100 Subject: IVGCVSW-5908 Update 21.05 Doxygen Documents Signed-off-by: mathad01 Change-Id: I95316d4fc5f9d10185492dc835bb2411c1daea7b --- ...armnn_1_1_shapes_are_broadcast_compatible.xhtml | 246 +++++++++++++++++++++ 1 file changed, 246 insertions(+) create mode 100644 21.05/structarmnn_1_1_shapes_are_broadcast_compatible.xhtml (limited to '21.05/structarmnn_1_1_shapes_are_broadcast_compatible.xhtml') diff --git a/21.05/structarmnn_1_1_shapes_are_broadcast_compatible.xhtml b/21.05/structarmnn_1_1_shapes_are_broadcast_compatible.xhtml new file mode 100644 index 0000000000..7e907877cf --- /dev/null +++ b/21.05/structarmnn_1_1_shapes_are_broadcast_compatible.xhtml @@ -0,0 +1,246 @@ + + + + + + + + + + + + + +ArmNN: ShapesAreBroadcastCompatible Struct Reference + + + + + + + + + + + + + + + + +
+
+ + + + ArmNN + + + +
+
+  21.05 +
+
+
+ + + + + + + +
+
+ +
+
+
+ +
+ +
+
+ + +
+ +
+ +
+
+Public Member Functions | +List of all members
+
+
ShapesAreBroadcastCompatible Struct Reference
+
+
+ +

#include <LayerSupportRules.hpp>

+
+Inheritance diagram for ShapesAreBroadcastCompatible:
+
+
+ + +Rule + +
+ + + + + + + + + +

+Public Member Functions

unsigned int CalcInputSize (const TensorShape &in, const TensorShape &out, unsigned int idx)
 
 ShapesAreBroadcastCompatible (const TensorInfo &in0, const TensorInfo &in1, const TensorInfo &out)
 
- Public Member Functions inherited from Rule
bool operator() () const
 
+ + + + +

+Additional Inherited Members

- Public Attributes inherited from Rule
bool m_Res = true
 
+

Detailed Description

+
+

Definition at line 154 of file LayerSupportRules.hpp.

+

Constructor & Destructor Documentation

+ +

◆ ShapesAreBroadcastCompatible()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + + + + + + + +
ShapesAreBroadcastCompatible (const TensorInfo & in0,
const TensorInfo & in1,
const TensorInfo & out 
)
+ 		+inline
+
+ +

Definition at line 163 of file LayerSupportRules.hpp.

+ +

References TensorShape::GetNumDimensions(), TensorInfo::GetShape(), and Rule::m_Res.

+
164  {
165  const TensorShape& shape0 = in0.GetShape();
166  const TensorShape& shape1 = in1.GetShape();
167  const TensorShape& outShape = out.GetShape();
168 
169  for (unsigned int i=0; i < outShape.GetNumDimensions() && m_Res; i++)
170  {
171  unsigned int sizeOut = outShape[i];
172  unsigned int sizeIn0 = CalcInputSize(shape0, outShape, i);
173  unsigned int sizeIn1 = CalcInputSize(shape1, outShape, i);
174 
175  m_Res &= ((sizeIn0 == sizeOut) || (sizeIn0 == 1)) &&
176  ((sizeIn1 == sizeOut) || (sizeIn1 == 1));
177  }
178  }
+
armnn::ShapesAreBroadcastCompatible::CalcInputSize
unsigned int CalcInputSize(const TensorShape &in, const TensorShape &out, unsigned int idx)
Definition: LayerSupportRules.hpp:156
+
+
+
+

Member Function Documentation

+ +

◆ CalcInputSize()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + + + + + + + +
unsigned int CalcInputSize (const TensorShape & in,
const TensorShape & out,
unsigned int idx 
)
+ 		+inline
+
+ +

Definition at line 156 of file LayerSupportRules.hpp.

+ +

References TensorShape::GetNumDimensions().

+
157  {
158  unsigned int offset = out.GetNumDimensions() - in.GetNumDimensions();
159  unsigned int sizeIn = (idx < offset) ? 1 : in[idx-offset];
160  return sizeIn;
161  }
+
+
+
The documentation for this struct was generated from the following file: +
+
+ + + + -- cgit v1.2.1