ArmNN
 20.05
Classes | Typedefs | Functions
armnnDeserializer Namespace Reference

Classes

struct  BindingPointInfo
 
class  Deserializer
 
class  IDeserializer
 

Typedefs

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

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::UnaryOperation ToUnaryOperation (armnnSerializer::UnaryOperation operation)
 
armnn::ResizeMethod ToResizeMethod (armnnSerializer::ResizeMethod method)
 
armnn::TensorInfo ToTensorInfo (Deserializer::TensorRawPtr tensorPtr)
 
armnn::ConstTensor ToConstTensor (Deserializer::ConstTensorRawPtr constTensorPtr)
 
const armnnSerializer::OriginsDescriptor * GetOriginsDescriptor (const armnnSerializer::SerializedGraph *graph, unsigned int layerIndex)
 

Typedef Documentation

◆ IDeserializerPtr

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

Definition at line 25 of file IDeserializer.hpp.

Function Documentation

◆ CheckShape()

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

Definition at line 164 of file Deserializer.cpp.

References TensorShape::GetNumDimensions().

Referenced by Deserializer::OutputShapeOfReshape(), and TfLiteParser::OutputShapeOfReshape().

165 {
166  const unsigned int actualSize = actual.GetNumDimensions();
167  if (actualSize != expected.size())
168  {
169  return false;
170  }
171 
172  for (unsigned int i = 0u; i < actualSize; i++)
173  {
174  if (actual[i] != static_cast<unsigned int>(expected[i]))
175  {
176  return false;
177  }
178  }
179 
180  return true;
181 }
armnn::TensorShape::GetNumDimensions
unsigned int GetNumDimensions() const
Definition: Tensor.hpp:43

◆ GetOriginsDescriptor()

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

Definition at line 1574 of file Deserializer.cpp.

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

1576 {
1577  auto layerType = graph->layers()->Get(layerIndex)->layer_type();
1578 
1579  switch (layerType)
1580  {
1581  case Layer::Layer_ConcatLayer:
1582  return graph->layers()->Get(layerIndex)->layer_as_ConcatLayer()->descriptor();
1583  case Layer::Layer_MergerLayer:
1584  return graph->layers()->Get(layerIndex)->layer_as_MergerLayer()->descriptor();
1585  default:
1586  throw armnn::Exception("unknown layer type, should be concat or merger");
1587  }
1588 }
armnn::Exception
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 406 of file Deserializer.cpp.

References armnn::Abs, armnn::BoundedReLu, armnn::Elu, armnn::HardSwish, armnn::LeakyReLu, armnn::Linear, armnn::ReLu, armnn::Sigmoid, armnn::Sqrt, armnn::Square, and armnn::TanH.

Referenced by Deserializer::GetNetworkOutputBindingInfo().

407 {
408  switch (function)
409  {
410  case armnnSerializer::ActivationFunction_Sigmoid:
411  return armnn::ActivationFunction::Sigmoid;
412  case armnnSerializer::ActivationFunction_TanH:
413  return armnn::ActivationFunction::TanH;
414  case armnnSerializer::ActivationFunction_Linear:
415  return armnn::ActivationFunction::Linear;
416  case armnnSerializer::ActivationFunction_ReLu:
417  return armnn::ActivationFunction::ReLu;
418  case armnnSerializer::ActivationFunction_BoundedReLu:
419  return armnn::ActivationFunction::BoundedReLu;
420  case armnnSerializer::ActivationFunction_LeakyReLu:
421  return armnn::ActivationFunction::LeakyReLu;
422  case armnnSerializer::ActivationFunction_Abs:
423  return armnn::ActivationFunction::Abs;
424  case armnnSerializer::ActivationFunction_Sqrt:
425  return armnn::ActivationFunction::Sqrt;
426  case armnnSerializer::ActivationFunction_Square:
427  return armnn::ActivationFunction::Square;
428  case armnnSerializer::ActivationFunction_Elu:
429  return armnn::ActivationFunction::Elu;
430  case armnnSerializer::ActivationFunction_HardSwish:
431  return armnn::ActivationFunction::HardSwish;
432  default:
433  return armnn::ActivationFunction::Sigmoid;
434  }
435 }
armnn::ActivationFunction::BoundedReLu
min(a, max(b, input)) ReLu1 & ReLu6.

◆ ToArgMinMaxFunction()

armnn::ArgMinMaxFunction armnnDeserializer::ToArgMinMaxFunction ( armnnSerializer::ArgMinMaxFunction  function)

Definition at line 437 of file Deserializer.cpp.

References armnn::Max, and armnn::Min.

Referenced by Deserializer::GetNetworkOutputBindingInfo().

438 {
439  switch (function)
440  {
441  case armnnSerializer::ArgMinMaxFunction::ArgMinMaxFunction_Max:
442  return armnn::ArgMinMaxFunction::Max;
443  case armnnSerializer::ArgMinMaxFunction::ArgMinMaxFunction_Min:
444  default:
445  return armnn::ArgMinMaxFunction::Min;
446  }
447 }

◆ ToComparisonOperation()

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

Definition at line 449 of file Deserializer.cpp.

References armnn::Equal, armnn::Greater, armnn::GreaterOrEqual, armnn::Less, armnn::LessOrEqual, and armnn::NotEqual.

Referenced by GetOriginsDescriptor().

450 {
451  switch (operation)
452  {
453  case armnnSerializer::ComparisonOperation::ComparisonOperation_Equal:
454  return armnn::ComparisonOperation::Equal;
455  case armnnSerializer::ComparisonOperation::ComparisonOperation_Greater:
456  return armnn::ComparisonOperation::Greater;
457  case armnnSerializer::ComparisonOperation::ComparisonOperation_GreaterOrEqual:
458  return armnn::ComparisonOperation::GreaterOrEqual;
459  case armnnSerializer::ComparisonOperation::ComparisonOperation_Less:
460  return armnn::ComparisonOperation::Less;
461  case armnnSerializer::ComparisonOperation::ComparisonOperation_LessOrEqual:
462  return armnn::ComparisonOperation::LessOrEqual;
463  case armnnSerializer::ComparisonOperation::ComparisonOperation_NotEqual:
464  default:
465  return armnn::ComparisonOperation::NotEqual;
466  }
467 }

◆ ToConstTensor()

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

Definition at line 578 of file Deserializer.cpp.

References CheckLocation::AsString(), CHECK_CONST_TENSOR_PTR, CHECK_CONST_TENSOR_SIZE, CHECK_LOCATION, TensorInfo::GetNumElements(), and ToTensorInfo().

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

579 {
580  CHECK_CONST_TENSOR_PTR(constTensorPtr);
581  armnn::TensorInfo tensorInfo = ToTensorInfo(constTensorPtr->info());
582 
583  switch (constTensorPtr->data_type())
584  {
585  case ConstTensorData_ByteData:
586  {
587  auto byteData = constTensorPtr->data_as_ByteData()->data();
588  CHECK_CONST_TENSOR_SIZE(byteData->size(), tensorInfo.GetNumElements());
589  return armnn::ConstTensor(tensorInfo, byteData->data());
590  }
591  case ConstTensorData_ShortData:
592  {
593  auto shortData = constTensorPtr->data_as_ShortData()->data();
594  CHECK_CONST_TENSOR_SIZE(shortData->size(), tensorInfo.GetNumElements());
595  return armnn::ConstTensor(tensorInfo, shortData->data());
596  }
597  case ConstTensorData_IntData:
598  {
599  auto intData = constTensorPtr->data_as_IntData()->data();
600  CHECK_CONST_TENSOR_SIZE(intData->size(), tensorInfo.GetNumElements());
601  return armnn::ConstTensor(tensorInfo, intData->data());
602  }
603  case ConstTensorData_LongData:
604  {
605  auto longData = constTensorPtr->data_as_LongData()->data();
606  CHECK_CONST_TENSOR_SIZE(longData->size(), tensorInfo.GetNumElements());
607  return armnn::ConstTensor(tensorInfo, longData->data());
608  }
609  default:
610  {
611  CheckLocation location = CHECK_LOCATION();
612  throw ParseException(
613  boost::str(boost::format("Unsupported data type %1% = %2%. %3%") %
614  constTensorPtr->data_type() %
615  EnumNameConstTensorData(constTensorPtr->data_type()) %
616  location.AsString()));
617  }
618  }
619 }
armnn::CheckLocation::AsString
std::string AsString() const
Definition: Exceptions.hpp:29
CHECK_CONST_TENSOR_SIZE
#define CHECK_CONST_TENSOR_SIZE(CONST_TENSOR_SIZE, TENSOR_SIZE)
Definition: Deserializer.cpp:151
armnnDeserializer::ToTensorInfo
armnn::TensorInfo ToTensorInfo(Deserializer::TensorRawPtr tensorPtr)
Definition: Deserializer.cpp:501
armnn::ConstTensor
A tensor defined by a TensorInfo (shape and data type) and an immutable backing store.
Definition: Tensor.hpp:199
CHECK_CONST_TENSOR_PTR
#define CHECK_CONST_TENSOR_PTR(TENSOR_PTR)
Definition: Deserializer.cpp:154
CHECK_LOCATION
#define CHECK_LOCATION()
Definition: Exceptions.hpp:192
armnn::TensorInfo::GetNumElements
unsigned int GetNumElements() const
Definition: Tensor.hpp:93

◆ ToDataLayout()

armnn::DataLayout armnnDeserializer::ToDataLayout ( armnnSerializer::DataLayout  dataLayout)

Definition at line 394 of file Deserializer.cpp.

References armnn::NCHW, and armnn::NHWC.

Referenced by Deserializer::GetNetworkOutputBindingInfo(), Deserializer::GetNormalizationDescriptor(), Deserializer::GetQLstmDescriptor(), and Deserializer::OutputShapeOfReshape().

395 {
396  switch (dataLayout)
397  {
398  case armnnSerializer::DataLayout::DataLayout_NHWC:
399  return armnn::DataLayout::NHWC;
400  case armnnSerializer::DataLayout::DataLayout_NCHW:
401  default:
402  return armnn::DataLayout::NCHW;
403  }
404 }

◆ ToResizeMethod()

armnn::ResizeMethod armnnDeserializer::ToResizeMethod ( armnnSerializer::ResizeMethod  method)

Definition at line 488 of file Deserializer.cpp.

References armnn::Bilinear, and armnn::NearestNeighbor.

Referenced by Deserializer::OutputShapeOfReshape().

489 {
490  switch (method)
491  {
492  case armnnSerializer::ResizeMethod_NearestNeighbor:
493  return armnn::ResizeMethod::NearestNeighbor;
494  case armnnSerializer::ResizeMethod_Bilinear:
495  return armnn::ResizeMethod::Bilinear;
496  default:
497  return armnn::ResizeMethod::NearestNeighbor;
498  }
499 }

◆ ToTensorInfo()

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

Definition at line 501 of file Deserializer.cpp.

References CheckLocation::AsString(), armnn::Boolean, CHECK_LOCATION, CHECK_TENSOR_PTR, armnn::Float16, armnn::Float32, armnn::QAsymmS8, armnn::QAsymmU8, armnn::QSymmS16, armnn::QSymmS8, and armnn::Signed32.

Referenced by armnnTfLiteParser::ComputeWrappedIndex(), TfLiteParser::CreateNetworkFromBinary(), OnnxParser::CreateNetworkFromString(), Deserializer::GetLstmDescriptor(), OnnxParser::GetNetworkInputBindingInfo(), TfLiteParser::GetNetworkInputBindingInfo(), Deserializer::GetNetworkOutputBindingInfo(), OnnxParser::GetNetworkOutputBindingInfo(), TfLiteParser::GetNetworkOutputBindingInfo(), Deserializer::GetNormalizationDescriptor(), GetOriginsDescriptor(), TfLiteParser::GetOutputTensorIds(), Deserializer::GetPoolingDescriptor(), Deserializer::GetQLstmDescriptor(), Deserializer::OutputShapeOfReshape(), TfLiteParser::OutputShapeOfReshape(), TfLiteParser::OutputShapeOfSqueeze(), TfLiteParser::TfLiteParser(), and ToConstTensor().

502 {
503  armnn::DataType type;
504  CHECK_TENSOR_PTR(tensorPtr);
505 
506  switch (tensorPtr->dataType())
507  {
508  case DataType_QAsymmS8:
509  type = armnn::DataType::QAsymmS8;
510  break;
511  case DataType_QSymmS8:
512  type = armnn::DataType::QSymmS8;
513  break;
514  case DataType_QuantisedAsymm8:
515  case DataType_QAsymmU8:
516  type = armnn::DataType::QAsymmU8;
517  break;
518  case DataType_QSymmS16:
519  case DataType_QuantisedSymm16:
520  type = armnn::DataType::QSymmS16;
521  break;
522  case DataType_Signed32:
523  type = armnn::DataType::Signed32;
524  break;
525  case DataType_Float32:
526  type = armnn::DataType::Float32;
527  break;
528  case DataType_Float16:
529  type = armnn::DataType::Float16;
530  break;
531  case DataType_Boolean:
532  type = armnn::DataType::Boolean;
533  break;
534  default:
535  {
536  CheckLocation location = CHECK_LOCATION();
537  throw ParseException(
538  boost::str(
539  boost::format("Unsupported data type %1% = %2%. %3%") %
540  tensorPtr->dataType() %
541  EnumNameDataType(tensorPtr->dataType()) %
542  location.AsString()));
543  }
544  }
545 
546 
547  auto dimensions = tensorPtr->dimensions();
548  unsigned int size = dimensions->size();
549  std::vector<unsigned int> outputDims(dimensions->begin(), dimensions->begin() + size);
550 
551  auto quantizationScales = tensorPtr->quantizationScales();
552 
553  if (quantizationScales)
554  {
555  unsigned int quantizationScalesSize = quantizationScales->size();
556  std::vector<float> scales(quantizationScales->begin(), quantizationScales->begin() + quantizationScalesSize);
557  unsigned int quantizationDim = tensorPtr->quantizationDim();
558  armnn::TensorInfo result(size,
559  outputDims.data(),
560  type,
561  scales,
562  quantizationDim);
563  return result;
564  }
565 
566  float quantizationScale = tensorPtr->quantizationScale();
567  int32_t quantizationOffset = tensorPtr->quantizationOffset();
568 
569  // two statements (on purpose) for easier debugging:
570  armnn::TensorInfo result(size,
571  outputDims.data(),
572  type,
573  quantizationScale,
574  quantizationOffset);
575  return result;
576 }
armnn::CheckLocation::AsString
std::string AsString() const
Definition: Exceptions.hpp:29
CHECK_TENSOR_PTR
#define CHECK_TENSOR_PTR(TENSOR_PTR)
Definition: Deserializer.cpp:148
armnn::DataType
DataType
Definition: Types.hpp:32
CHECK_LOCATION
#define CHECK_LOCATION()
Definition: Exceptions.hpp:192

◆ ToUnaryOperation()

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

Definition at line 469 of file Deserializer.cpp.

References armnn::Abs, armnn::Exp, armnn::Neg, armnn::Rsqrt, and armnn::Sqrt.

Referenced by GetOriginsDescriptor().

470 {
471  switch (operation)
472  {
473  case armnnSerializer::UnaryOperation::UnaryOperation_Abs:
474  return armnn::UnaryOperation::Abs;
475  case armnnSerializer::UnaryOperation::UnaryOperation_Rsqrt:
476  return armnn::UnaryOperation::Rsqrt;
477  case armnnSerializer::UnaryOperation::UnaryOperation_Sqrt:
478  return armnn::UnaryOperation::Sqrt;
479  case armnnSerializer::UnaryOperation::UnaryOperation_Exp:
480  return armnn::UnaryOperation::Exp;
481  case armnnSerializer::UnaryOperation::UnaryOperation_Neg:
482  return armnn::UnaryOperation::Neg;
483  default:
484  throw armnn::InvalidArgumentException("Unary operation unknown");
485  }
486 }