ArmNN
NotReleased
|
A ViewsDescriptor for the SplitterLayer. Descriptor to configure the splitting process. Number of Views must be equal to the number of outputs, and their order must match - e.g. first view corresponds to the first output, second view to the second output, etc. More...
#include <Descriptors.hpp>
Public Member Functions | |
ViewsDescriptor (uint32_t numViews, uint32_t numDimensions=4) | |
ViewsDescriptor (const ViewsDescriptor &other) | |
ViewsDescriptor () | |
ViewsDescriptor (ViewsDescriptor &&other) | |
~ViewsDescriptor () | |
ViewsDescriptor & | operator= (ViewsDescriptor rhs) |
bool | operator== (const ViewsDescriptor &rhs) const |
Status | SetViewOriginCoord (uint32_t view, uint32_t coord, uint32_t value) |
Status | SetViewSize (uint32_t view, uint32_t coord, uint32_t value) |
Set the size of the views. The arguments are: view, dimension, value. If the view is greater than or equal to GetNumViews(), then the view argument is out of range. If the coord is greater than or equal to GetNumDimensions(), then the coord argument is out of range. More... | |
uint32_t | GetNumViews () const |
Get the number of views. More... | |
uint32_t | GetNumDimensions () const |
Get the number of dimensions. More... | |
const uint32_t * | GetViewOrigin (uint32_t idx) const |
Get the view origin at the int value idx. More... | |
const uint32_t * | GetViewSizes (uint32_t idx) const |
Get the view sizes at the int value idx. More... | |
const OriginsDescriptor & | GetOrigins () const |
Get the View Origins. More... | |
Friends | |
void | swap (ViewsDescriptor &first, ViewsDescriptor &second) |
Swap the ViewsDescriptor value first and second. More... | |
A ViewsDescriptor for the SplitterLayer. Descriptor to configure the splitting process. Number of Views must be equal to the number of outputs, and their order must match - e.g. first view corresponds to the first output, second view to the second output, etc.
Definition at line 190 of file Descriptors.hpp.
ViewsDescriptor | ( | uint32_t | numViews, |
uint32_t | numDimensions = 4 |
||
) |
Definition at line 213 of file Descriptors.cpp.
References ViewsDescriptor::GetNumDimensions(), and ViewsDescriptor::GetNumViews().
ViewsDescriptor | ( | const ViewsDescriptor & | other | ) |
Definition at line 227 of file Descriptors.cpp.
References ViewsDescriptor::GetNumDimensions(), and ViewsDescriptor::GetNumViews().
ViewsDescriptor | ( | ) |
Definition at line 208 of file Descriptors.cpp.
ViewsDescriptor | ( | ViewsDescriptor && | other | ) |
Definition at line 242 of file Descriptors.cpp.
References ViewsDescriptor::swap.
~ViewsDescriptor | ( | ) |
Definition at line 248 of file Descriptors.cpp.
References ViewsDescriptor::GetNumDimensions(), and ViewsDescriptor::GetNumViews().
uint32_t GetNumDimensions | ( | ) | const |
Get the number of dimensions.
Definition at line 292 of file Descriptors.cpp.
References OriginsDescriptor::GetNumDimensions().
Referenced by ClSplitterWorkload::ClSplitterWorkload(), armnn::ComputeSplitAxis(), SplitterLayer::CreateWorkload(), SplitterLayer::InferOutputShapes(), ClLayerSupport::IsSplitterSupported(), NeonLayerSupport::IsSplitterSupported(), NeonSplitterWorkload::NeonSplitterWorkload(), ViewsDescriptor::operator==(), StringifyLayerParameters< ViewsDescriptor >::Serialize(), ViewsDescriptor::SetViewSize(), SplitterLayer::ValidateTensorShapesFromInputs(), ViewsDescriptor::ViewsDescriptor(), SerializerVisitor::VisitSplitterLayer(), and ViewsDescriptor::~ViewsDescriptor().
uint32_t GetNumViews | ( | ) | const |
Get the number of views.
Definition at line 287 of file Descriptors.cpp.
References OriginsDescriptor::GetNumViews().
Referenced by armnn::ComputeSplitAxis(), SplitterLayer::CreateWorkload(), SplitterLayer::InferOutputShapes(), ViewsDescriptor::operator==(), StringifyLayerParameters< ViewsDescriptor >::Serialize(), ViewsDescriptor::SetViewSize(), SplitterLayer::ValidateTensorShapesFromInputs(), ViewsDescriptor::ViewsDescriptor(), SerializerVisitor::VisitSplitterLayer(), and ViewsDescriptor::~ViewsDescriptor().
const OriginsDescriptor & GetOrigins | ( | ) | const |
Get the View Origins.
Definition at line 337 of file Descriptors.cpp.
Referenced by StringifyLayerParameters< ViewsDescriptor >::Serialize(), and SerializerVisitor::VisitSplitterLayer().
const uint32_t * GetViewOrigin | ( | uint32_t | idx | ) | const |
Get the view origin at the int value idx.
Definition at line 297 of file Descriptors.cpp.
References OriginsDescriptor::GetViewOrigin().
Referenced by SplitterLayer::CreateWorkload(), ViewsDescriptor::operator==(), and SerializerVisitor::VisitSplitterLayer().
const uint32_t * GetViewSizes | ( | uint32_t | idx | ) | const |
Get the view sizes at the int value idx.
Definition at line 332 of file Descriptors.cpp.
Referenced by armnn::ComputeSplitAxis(), SplitterLayer::InferOutputShapes(), ViewsDescriptor::operator==(), StringifyLayerParameters< ViewsDescriptor >::Serialize(), SplitterLayer::ValidateTensorShapesFromInputs(), and SerializerVisitor::VisitSplitterLayer().
ViewsDescriptor & operator= | ( | ViewsDescriptor | rhs | ) |
Definition at line 260 of file Descriptors.cpp.
References ViewsDescriptor::swap.
bool operator== | ( | const ViewsDescriptor & | rhs | ) | const |
Definition at line 266 of file Descriptors.cpp.
References ViewsDescriptor::GetNumDimensions(), ViewsDescriptor::GetNumViews(), ViewsDescriptor::GetViewOrigin(), and ViewsDescriptor::GetViewSizes().
Status SetViewOriginCoord | ( | uint32_t | view, |
uint32_t | coord, | ||
uint32_t | value | ||
) |
Set the view origin coordinates. The arguments are: view, dimension, value. If the view is greater than or equal to GetNumViews(), then the view argument is out of range. If the coord is greater than or equal to GetNumDimensions(), then the coord argument is out of range.
Definition at line 302 of file Descriptors.cpp.
References OriginsDescriptor::SetViewOriginCoord().
Referenced by CaffeParserBase::AddConvLayerWithSplits(), BOOST_AUTO_TEST_CASE(), Deserializer::GetNormalizationDescriptor(), TfLiteParser::OutputShapeOfReshape(), and armnnTfParser::OutputShapeOfSqueeze().
Status SetViewSize | ( | uint32_t | view, |
uint32_t | coord, | ||
uint32_t | value | ||
) |
Set the size of the views. The arguments are: view, dimension, value. If the view is greater than or equal to GetNumViews(), then the view argument is out of range. If the coord is greater than or equal to GetNumDimensions(), then the coord argument is out of range.
Definition at line 307 of file Descriptors.cpp.
References ARMNN_LOG, armnn::error, armnn::Failure, ViewsDescriptor::GetNumDimensions(), ViewsDescriptor::GetNumViews(), and armnn::Success.
Referenced by CaffeParserBase::AddConvLayerWithSplits(), BOOST_AUTO_TEST_CASE(), Deserializer::GetNormalizationDescriptor(), TfLiteParser::OutputShapeOfReshape(), and armnnTfParser::OutputShapeOfSqueeze().
|
friend |
Swap the ViewsDescriptor value first and second.
Definition at line 351 of file Descriptors.cpp.
Referenced by ViewsDescriptor::operator=(), armnn::swap(), and ViewsDescriptor::ViewsDescriptor().