ArmNN
 21.02
armnnDeserializer Namespace Reference

Classes

struct  BindingPointInfo
 
class  IDeserializer
 

Typedefs

using IDeserializerPtr = std::unique_ptr< IDeserializer, void(*)(IDeserializer *parser)>
 
using ConstTensorRawPtr = const armnnSerializer::ConstTensor *
 
using GraphPtr = const armnnSerializer::SerializedGraph *
 
using TensorRawPtr = const armnnSerializer::TensorInfo *
 
using PoolingDescriptor = const armnnSerializer::Pooling2dDescriptor *
 
using NormalizationDescriptorPtr = const armnnSerializer::NormalizationDescriptor *
 
using LstmDescriptorPtr = const armnnSerializer::LstmDescriptor *
 
using LstmInputParamsPtr = const armnnSerializer::LstmInputParams *
 
using QLstmDescriptorPtr = const armnnSerializer::QLstmDescriptor *
 
using QunatizedLstmInputParamsPtr = const armnnSerializer::QuantizedLstmInputParams *
 
using TensorRawPtrVector = std::vector< TensorRawPtr >
 
using LayerRawPtr = const armnnSerializer::LayerBase *
 
using LayerBaseRawPtr = const armnnSerializer::LayerBase *
 
using LayerBaseRawPtrVector = std::vector< LayerBaseRawPtr >
 

Functions

bool CheckShape (const armnn::TensorShape &actual, const std::vector< uint32_t > &expected)
 
armnn::DataLayout ToDataLayout (armnnSerializer::DataLayout dataLayout)
 
armnn::ActivationFunction ToActivationFunction (armnnSerializer::ActivationFunction function)
 
armnn::ArgMinMaxFunction ToArgMinMaxFunction (armnnSerializer::ArgMinMaxFunction function)
 
armnn::ComparisonOperation ToComparisonOperation (armnnSerializer::ComparisonOperation operation)
 
armnn::ReduceOperation ToReduceOperation (armnnSerializer::ReduceOperation operation)
 
armnn::LogicalBinaryOperation ToLogicalBinaryOperation (armnnSerializer::LogicalBinaryOperation operation)
 
armnn::UnaryOperation ToUnaryOperation (armnnSerializer::UnaryOperation operation)
 
armnn::ResizeMethod ToResizeMethod (armnnSerializer::ResizeMethod method)
 
armnn::TensorInfo ToTensorInfo (TensorRawPtr tensorPtr)
 
armnn::ConstTensor ToConstTensor (ConstTensorRawPtr constTensorPtr)
 
const armnnSerializer::OriginsDescriptor * GetOriginsDescriptor (const armnnSerializer::SerializedGraph *graph, unsigned int layerIndex)
 

Typedef Documentation

◆ ConstTensorRawPtr

using ConstTensorRawPtr = const armnnSerializer::ConstTensor *

Definition at line 18 of file Deserializer.hpp.

◆ GraphPtr

using GraphPtr = const armnnSerializer::SerializedGraph *

Definition at line 19 of file Deserializer.hpp.

◆ IDeserializerPtr

using IDeserializerPtr = std::unique_ptr<IDeserializer, void(*)(IDeserializer* parser)>

Definition at line 25 of file IDeserializer.hpp.

◆ LayerBaseRawPtr

using LayerBaseRawPtr = const armnnSerializer::LayerBase *

Definition at line 29 of file Deserializer.hpp.

◆ LayerBaseRawPtrVector

using LayerBaseRawPtrVector = std::vector<LayerBaseRawPtr>

Definition at line 30 of file Deserializer.hpp.

◆ LayerRawPtr

using LayerRawPtr = const armnnSerializer::LayerBase *

Definition at line 28 of file Deserializer.hpp.

◆ LstmDescriptorPtr

using LstmDescriptorPtr = const armnnSerializer::LstmDescriptor *

Definition at line 23 of file Deserializer.hpp.

◆ LstmInputParamsPtr

using LstmInputParamsPtr = const armnnSerializer::LstmInputParams *

Definition at line 24 of file Deserializer.hpp.

◆ NormalizationDescriptorPtr

using NormalizationDescriptorPtr = const armnnSerializer::NormalizationDescriptor *

Definition at line 22 of file Deserializer.hpp.

◆ PoolingDescriptor

using PoolingDescriptor = const armnnSerializer::Pooling2dDescriptor *

Definition at line 21 of file Deserializer.hpp.

◆ QLstmDescriptorPtr

using QLstmDescriptorPtr = const armnnSerializer::QLstmDescriptor *

Definition at line 25 of file Deserializer.hpp.

◆ QunatizedLstmInputParamsPtr

using QunatizedLstmInputParamsPtr = const armnnSerializer::QuantizedLstmInputParams *

Definition at line 26 of file Deserializer.hpp.

◆ TensorRawPtr

using TensorRawPtr = const armnnSerializer::TensorInfo *

Definition at line 20 of file Deserializer.hpp.

◆ TensorRawPtrVector

using TensorRawPtrVector = std::vector<TensorRawPtr>

Definition at line 27 of file Deserializer.hpp.

Function Documentation

◆ CheckShape()

bool armnnDeserializer::CheckShape ( const armnn::TensorShape actual,
const std::vector< uint32_t > &  expected 
)

Definition at line 187 of file Deserializer.cpp.

References TensorShape::GetNumDimensions().

Referenced by IDeserializer::DeserializerImpl::OutputShapeOfReshape(), and TfLiteParserImpl::OutputShapeOfReshape().

188 {
189  const unsigned int actualSize = actual.GetNumDimensions();
190  if (actualSize != expected.size())
191  {
192  return false;
193  }
194 
195  for (unsigned int i = 0u; i < actualSize; i++)
196  {
197  if (actual[i] != static_cast<unsigned int>(expected[i]))
198  {
199  return false;
200  }
201  }
202 
203  return true;
204 }
unsigned int GetNumDimensions() const
Function that returns the tensor rank.
Definition: Tensor.cpp:174

◆ GetOriginsDescriptor()

const armnnSerializer::OriginsDescriptor* armnnDeserializer::GetOriginsDescriptor ( const armnnSerializer::SerializedGraph *  graph,
unsigned int  layerIndex 
)

Definition at line 1687 of file Deserializer.cpp.

References CHECK_LAYERS, CHECK_LOCATION, CHECK_VALID_SIZE, IDeserializer::DeserializerImpl::GetInputs(), IDeserializer::DeserializerImpl::GetLayerName(), IDeserializer::DeserializerImpl::GetOutputs(), IConnectableLayer::GetOutputSlot(), armnnSerializer::Layer_ConcatLayer, armnnSerializer::Layer_MergerLayer, FullyConnectedDescriptor::m_BiasEnabled, ComparisonDescriptor::m_Operation, ElementwiseUnaryDescriptor::m_Operation, FullyConnectedDescriptor::m_TransposeWeightMatrix, IOutputSlot::SetTensorInfo(), ToComparisonOperation(), ToConstTensor(), ToTensorInfo(), and ToUnaryOperation().

1689 {
1690  auto layerType = graph->layers()->Get(layerIndex)->layer_type();
1691 
1692  switch (layerType)
1693  {
1695  return graph->layers()->Get(layerIndex)->layer_as_ConcatLayer()->descriptor();
1697  return graph->layers()->Get(layerIndex)->layer_as_MergerLayer()->descriptor();
1698  default:
1699  throw armnn::Exception("unknown layer type, should be concat or merger");
1700  }
1701 }
Base class for all ArmNN exceptions so that users can filter to just those.
Definition: Exceptions.hpp:46

◆ ToActivationFunction()

armnn::ActivationFunction armnnDeserializer::ToActivationFunction ( armnnSerializer::ActivationFunction  function)

Definition at line 441 of file Deserializer.cpp.

References armnn::Abs, armnnSerializer::ActivationFunction_Abs, armnnSerializer::ActivationFunction_BoundedReLu, armnnSerializer::ActivationFunction_Elu, armnnSerializer::ActivationFunction_HardSwish, armnnSerializer::ActivationFunction_LeakyReLu, armnnSerializer::ActivationFunction_Linear, armnnSerializer::ActivationFunction_ReLu, armnnSerializer::ActivationFunction_Sigmoid, armnnSerializer::ActivationFunction_Sqrt, armnnSerializer::ActivationFunction_Square, armnnSerializer::ActivationFunction_TanH, armnn::BoundedReLu, armnn::Elu, armnn::HardSwish, armnn::LeakyReLu, armnn::Linear, armnn::ReLu, armnn::Sigmoid, armnn::Sqrt, armnn::Square, and armnn::TanH.

Referenced by IDeserializer::DeserializerImpl::GetNetworkOutputBindingInfo().

442 {
443  switch (function)
444  {
467  default:
469  }
470 }
min(a, max(b, input)) ReLu1 & ReLu6.

◆ ToArgMinMaxFunction()

◆ ToComparisonOperation()

armnn::ComparisonOperation armnnDeserializer::ToComparisonOperation ( armnnSerializer::ComparisonOperation  operation)

Definition at line 484 of file Deserializer.cpp.

References armnnSerializer::ComparisonOperation_Equal, armnnSerializer::ComparisonOperation_Greater, armnnSerializer::ComparisonOperation_GreaterOrEqual, armnnSerializer::ComparisonOperation_Less, armnnSerializer::ComparisonOperation_LessOrEqual, armnnSerializer::ComparisonOperation_NotEqual, armnn::Equal, armnn::Greater, armnn::GreaterOrEqual, armnn::Less, armnn::LessOrEqual, and armnn::NotEqual.

Referenced by GetOriginsDescriptor().

485 {
486  switch (operation)
487  {
499  default:
501  }
502 }

◆ ToConstTensor()

armnn::ConstTensor armnnDeserializer::ToConstTensor ( ConstTensorRawPtr  constTensorPtr)

Definition at line 671 of file Deserializer.cpp.

References CheckLocation::AsString(), CHECK_CONST_TENSOR_PTR, CHECK_CONST_TENSOR_SIZE, CHECK_LOCATION, armnnSerializer::ConstTensorData_ByteData, armnnSerializer::ConstTensorData_IntData, armnnSerializer::ConstTensorData_LongData, armnnSerializer::ConstTensorData_ShortData, armnnSerializer::EnumNameConstTensorData(), TensorInfo::GetNumElements(), and ToTensorInfo().

Referenced by IDeserializer::DeserializerImpl::GetLstmDescriptor(), IDeserializer::DeserializerImpl::GetNetworkOutputBindingInfo(), GetOriginsDescriptor(), and IDeserializer::DeserializerImpl::GetQLstmDescriptor().

672 {
673  CHECK_CONST_TENSOR_PTR(constTensorPtr);
674  armnn::TensorInfo tensorInfo = ToTensorInfo(constTensorPtr->info());
675 
676  switch (constTensorPtr->data_type())
677  {
679  {
680  auto byteData = constTensorPtr->data_as_ByteData()->data();
681  CHECK_CONST_TENSOR_SIZE(byteData->size(), tensorInfo.GetNumElements());
682  return armnn::ConstTensor(tensorInfo, byteData->data());
683  }
685  {
686  auto shortData = constTensorPtr->data_as_ShortData()->data();
687  CHECK_CONST_TENSOR_SIZE(shortData->size(), tensorInfo.GetNumElements());
688  return armnn::ConstTensor(tensorInfo, shortData->data());
689  }
691  {
692  auto intData = constTensorPtr->data_as_IntData()->data();
693  CHECK_CONST_TENSOR_SIZE(intData->size(), tensorInfo.GetNumElements());
694  return armnn::ConstTensor(tensorInfo, intData->data());
695  }
697  {
698  auto longData = constTensorPtr->data_as_LongData()->data();
699  CHECK_CONST_TENSOR_SIZE(longData->size(), tensorInfo.GetNumElements());
700  return armnn::ConstTensor(tensorInfo, longData->data());
701  }
702  default:
703  {
704  CheckLocation location = CHECK_LOCATION();
705  throw ParseException(fmt::format("Unsupported data type {0} = {1}. {2}",
706  constTensorPtr->data_type(),
707  EnumNameConstTensorData(constTensorPtr->data_type()),
708  location.AsString()));
709  }
710  }
711 }
std::string AsString() const
Definition: Exceptions.hpp:29
const char * EnumNameConstTensorData(ConstTensorData e)
#define CHECK_CONST_TENSOR_SIZE(CONST_TENSOR_SIZE, TENSOR_SIZE)
A tensor defined by a TensorInfo (shape and data type) and an immutable backing store.
Definition: Tensor.hpp:314
#define CHECK_CONST_TENSOR_PTR(TENSOR_PTR)
#define CHECK_LOCATION()
Definition: Exceptions.hpp:197
armnn::TensorInfo ToTensorInfo(TensorRawPtr tensorPtr)
unsigned int GetNumElements() const
Definition: Tensor.hpp:192

◆ ToDataLayout()

◆ ToLogicalBinaryOperation()

◆ ToReduceOperation()

◆ ToResizeMethod()

◆ ToTensorInfo()

armnn::TensorInfo armnnDeserializer::ToTensorInfo ( TensorRawPtr  tensorPtr)

Definition at line 568 of file Deserializer.cpp.

References CheckLocation::AsString(), armnn::Boolean, CHECK_LOCATION, CHECK_TENSOR_PTR, armnnSerializer::DataType_Boolean, armnnSerializer::DataType_Float16, armnnSerializer::DataType_Float32, armnnSerializer::DataType_QAsymmS8, armnnSerializer::DataType_QAsymmU8, armnnSerializer::DataType_QSymmS16, armnnSerializer::DataType_QSymmS8, armnnSerializer::DataType_QuantisedAsymm8, armnnSerializer::DataType_QuantisedSymm16, armnnSerializer::DataType_Signed32, armnnSerializer::EnumNameDataType(), armnn::Float16, armnn::Float32, armnn::MaxNumOfTensorDimensions, armnn::QAsymmS8, armnn::QAsymmU8, armnn::QSymmS16, armnn::QSymmS8, armnn::Scalar, and armnn::Signed32.

Referenced by armnnTfLiteParser::ComputeWrappedIndex(), TfLiteParserImpl::CreateNetworkFromBinary(), OnnxParserImpl::CreateNetworkFromString(), IDeserializer::DeserializerImpl::GetLstmDescriptor(), OnnxParserImpl::GetNetworkInputBindingInfo(), TfLiteParserImpl::GetNetworkInputBindingInfo(), IDeserializer::DeserializerImpl::GetNetworkOutputBindingInfo(), OnnxParserImpl::GetNetworkOutputBindingInfo(), TfLiteParserImpl::GetNetworkOutputBindingInfo(), IDeserializer::DeserializerImpl::GetNormalizationDescriptor(), GetOriginsDescriptor(), TfLiteParserImpl::GetOutputTensorIds(), IDeserializer::DeserializerImpl::GetPoolingDescriptor(), IDeserializer::DeserializerImpl::GetQLstmDescriptor(), IDeserializer::DeserializerImpl::OutputShapeOfReshape(), TfLiteParserImpl::OutputShapeOfReshape(), TfLiteParserImpl::OutputShapeOfSqueeze(), and ToConstTensor().

569 {
570  armnn::DataType type;
571  CHECK_TENSOR_PTR(tensorPtr);
572 
573  switch (tensorPtr->dataType())
574  {
575  case DataType_QAsymmS8:
577  break;
578  case DataType_QSymmS8:
580  break;
582  case DataType_QAsymmU8:
584  break;
585  case DataType_QSymmS16:
588  break;
589  case DataType_Signed32:
591  break;
592  case DataType_Float32:
594  break;
595  case DataType_Float16:
597  break;
598  case DataType_Boolean:
600  break;
601  default:
602  {
603  CheckLocation location = CHECK_LOCATION();
604  throw ParseException(fmt::format("Unsupported data type {0} = {1}. {2}",
605  tensorPtr->dataType(),
606  EnumNameDataType(tensorPtr->dataType()),
607  location.AsString()));
608  }
609  }
610 
611  float quantizationScale = tensorPtr->quantizationScale();
612  int32_t quantizationOffset = tensorPtr->quantizationOffset();
613 
614  if (tensorPtr->dimensionality() == static_cast<unsigned int>(Dimensionality::Scalar))
615  {
617  type,
618  quantizationScale,
619  quantizationOffset);
620  }
621  else if (tensorPtr->dimensionality() == static_cast<unsigned int>(Dimensionality::NotSpecified))
622  {
623  armnn::TensorInfo result(TensorShape{Dimensionality::NotSpecified},
624  type,
625  quantizationScale,
626  quantizationOffset);
627  return result;
628  }
629 
630  auto dimensions = tensorPtr->dimensions();
631  unsigned int size = dimensions->size();
632  std::vector<unsigned int> outputDims(dimensions->begin(), dimensions->begin() + size);
633  bool dimensionsSpecificity[armnn::MaxNumOfTensorDimensions];
634  std::fill_n(dimensionsSpecificity, armnn::MaxNumOfTensorDimensions, true);
635  // For backwards compatibility check if the dimensionSpecificity vector is present first.
636  // The default is to have dimensionSpecificity set to all true's anyway.
637  if (tensorPtr->dimensionSpecificity() != nullptr)
638  {
639  auto dimensionSpecificity = tensorPtr->dimensionSpecificity();
640  size = dimensionSpecificity->size();
641  for (unsigned int i = 0; i < size; ++i)
642  {
643  dimensionsSpecificity[i] = dimensionSpecificity->Get(i);
644  }
645  }
646  // Construct a TensorShape
647  TensorShape shape(size, outputDims.data(), dimensionsSpecificity);
648 
649  auto quantizationScales = tensorPtr->quantizationScales();
650  if (quantizationScales)
651  {
652  unsigned int quantizationScalesSize = quantizationScales->size();
653  std::vector<float> scales(quantizationScales->begin(), quantizationScales->begin() + quantizationScalesSize);
654  unsigned int quantizationDim = tensorPtr->quantizationDim();
655  armnn::TensorInfo result(shape,
656  type,
657  scales,
658  quantizationDim);
659  return result;
660  }
661 
662  // two statements (on purpose) for easier debugging:
663  armnn::TensorInfo result(shape,
664  type,
665  quantizationScale,
666  quantizationOffset);
667 
668  return result;
669 }
std::string AsString() const
Definition: Exceptions.hpp:29
#define CHECK_TENSOR_PTR(TENSOR_PTR)
DataType
Definition: Types.hpp:32
#define CHECK_LOCATION()
Definition: Exceptions.hpp:197
const char * EnumNameDataType(DataType e)
constexpr unsigned int MaxNumOfTensorDimensions
Definition: Types.hpp:18

◆ ToUnaryOperation()

armnn::UnaryOperation armnnDeserializer::ToUnaryOperation ( armnnSerializer::UnaryOperation  operation)

Definition at line 534 of file Deserializer.cpp.

References armnn::Abs, armnn::Exp, armnn::LogicalNot, armnn::Neg, armnn::Rsqrt, armnn::Sqrt, armnnSerializer::UnaryOperation_Abs, armnnSerializer::UnaryOperation_Exp, armnnSerializer::UnaryOperation_LogicalNot, armnnSerializer::UnaryOperation_Neg, armnnSerializer::UnaryOperation_Rsqrt, and armnnSerializer::UnaryOperation_Sqrt.

Referenced by GetOriginsDescriptor().