ArmNN
 21.05
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 1715 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, FullyConnectedDescriptor::m_ConstantWeights, ComparisonDescriptor::m_Operation, ElementwiseUnaryDescriptor::m_Operation, FullyConnectedDescriptor::m_TransposeWeightMatrix, IOutputSlot::SetTensorInfo(), ToComparisonOperation(), ToConstTensor(), ToTensorInfo(), and ToUnaryOperation().

1717 {
1718  auto layerType = graph->layers()->Get(layerIndex)->layer_type();
1719 
1720  switch (layerType)
1721  {
1723  return graph->layers()->Get(layerIndex)->layer_as_ConcatLayer()->descriptor();
1725  return graph->layers()->Get(layerIndex)->layer_as_MergerLayer()->descriptor();
1726  default:
1727  throw armnn::Exception("unknown layer type, should be concat or merger");
1728  }
1729 }
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 444 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().

445 {
446  switch (function)
447  {
470  default:
472  }
473 }
min(a, max(b, input)) ReLu1 & ReLu6.

◆ ToArgMinMaxFunction()

◆ ToComparisonOperation()

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

Definition at line 487 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().

488 {
489  switch (operation)
490  {
502  default:
504  }
505 }

◆ ToConstTensor()

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

Definition at line 677 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().

678 {
679  CHECK_CONST_TENSOR_PTR(constTensorPtr);
680  armnn::TensorInfo tensorInfo = ToTensorInfo(constTensorPtr->info());
681 
682  switch (constTensorPtr->data_type())
683  {
685  {
686  auto byteData = constTensorPtr->data_as_ByteData()->data();
687  CHECK_CONST_TENSOR_SIZE(byteData->size(), tensorInfo.GetNumElements());
688  return armnn::ConstTensor(tensorInfo, byteData->data());
689  }
691  {
692  auto shortData = constTensorPtr->data_as_ShortData()->data();
693  CHECK_CONST_TENSOR_SIZE(shortData->size(), tensorInfo.GetNumElements());
694  return armnn::ConstTensor(tensorInfo, shortData->data());
695  }
697  {
698  auto intData = constTensorPtr->data_as_IntData()->data();
699  CHECK_CONST_TENSOR_SIZE(intData->size(), tensorInfo.GetNumElements());
700  return armnn::ConstTensor(tensorInfo, intData->data());
701  }
703  {
704  auto longData = constTensorPtr->data_as_LongData()->data();
705  CHECK_CONST_TENSOR_SIZE(longData->size(), tensorInfo.GetNumElements());
706  return armnn::ConstTensor(tensorInfo, longData->data());
707  }
708  default:
709  {
710  CheckLocation location = CHECK_LOCATION();
711  throw ParseException(fmt::format("Unsupported data type {0} = {1}. {2}",
712  constTensorPtr->data_type(),
713  EnumNameConstTensorData(constTensorPtr->data_type()),
714  location.AsString()));
715  }
716  }
717 }
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 571 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::DataType_Signed64, armnnSerializer::EnumNameDataType(), armnn::Float16, armnn::Float32, armnn::MaxNumOfTensorDimensions, armnn::QAsymmS8, armnn::QAsymmU8, armnn::QSymmS16, armnn::QSymmS8, armnn::Scalar, armnn::Signed32, and armnn::Signed64.

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().

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

◆ ToUnaryOperation()

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

Definition at line 537 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().