ArmNN
 24.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 Pooling2dDescriptor = const armnnSerializer::Pooling2dDescriptor *
 
using Pooling3dDescriptor = const armnnSerializer::Pooling3dDescriptor *
 
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 >
 
using UnidirectionalSequenceLstmDescriptorPtr = const armnnSerializer::UnidirectionalSequenceLstmDescriptor *
 

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::ScatterNdFunction ToScatterNdFunction (armnnSerializer::ScatterNdFunction function)
 
armnn::ComparisonOperation ToComparisonOperation (armnnSerializer::ComparisonOperation operation)
 
armnn::ReduceOperation ToReduceOperation (armnnSerializer::ReduceOperation operation)
 
armnn::LogicalBinaryOperation ToLogicalBinaryOperation (armnnSerializer::LogicalBinaryOperation operation)
 
armnn::BinaryOperation ToElementwiseBinaryOperation (armnnSerializer::BinaryOperation operation)
 
armnn::UnaryOperation ToElementwiseUnaryOperation (armnnSerializer::UnaryOperation operation)
 
armnn::PaddingMode ToPaddingMode (armnnSerializer::PaddingMode paddingMode)
 
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 30 of file Deserializer.hpp.

◆ LayerBaseRawPtrVector

using LayerBaseRawPtrVector = std::vector<LayerBaseRawPtr>

Definition at line 31 of file Deserializer.hpp.

◆ LayerRawPtr

using LayerRawPtr = const armnnSerializer::LayerBase *

Definition at line 29 of file Deserializer.hpp.

◆ LstmDescriptorPtr

using LstmDescriptorPtr = const armnnSerializer::LstmDescriptor *

Definition at line 24 of file Deserializer.hpp.

◆ LstmInputParamsPtr

using LstmInputParamsPtr = const armnnSerializer::LstmInputParams *

Definition at line 25 of file Deserializer.hpp.

◆ NormalizationDescriptorPtr

using NormalizationDescriptorPtr = const armnnSerializer::NormalizationDescriptor *

Definition at line 23 of file Deserializer.hpp.

◆ Pooling2dDescriptor

using Pooling2dDescriptor = const armnnSerializer::Pooling2dDescriptor *

Definition at line 21 of file Deserializer.hpp.

◆ Pooling3dDescriptor

using Pooling3dDescriptor = const armnnSerializer::Pooling3dDescriptor *

Definition at line 22 of file Deserializer.hpp.

◆ QLstmDescriptorPtr

using QLstmDescriptorPtr = const armnnSerializer::QLstmDescriptor *

Definition at line 26 of file Deserializer.hpp.

◆ QunatizedLstmInputParamsPtr

using QunatizedLstmInputParamsPtr = const armnnSerializer::QuantizedLstmInputParams *

Definition at line 27 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 28 of file Deserializer.hpp.

◆ UnidirectionalSequenceLstmDescriptorPtr

using UnidirectionalSequenceLstmDescriptorPtr = const armnnSerializer::UnidirectionalSequenceLstmDescriptor *

Definition at line 32 of file Deserializer.hpp.

Function Documentation

◆ CheckShape()

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

Definition at line 188 of file Deserializer.cpp.

189 {
190  const unsigned int actualSize = actual.GetNumDimensions();
191  if (actualSize != expected.size())
192  {
193  return false;
194  }
195 
196  for (unsigned int i = 0u; i < actualSize; i++)
197  {
198  if (actual[i] != static_cast<unsigned int>(expected[i]))
199  {
200  return false;
201  }
202  }
203 
204  return true;
205 }

References TensorShape::GetNumDimensions().

◆ GetOriginsDescriptor()

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

Definition at line 2055 of file Deserializer.cpp.

2057 {
2058  auto layerType = graph->layers()->Get(layerIndex)->layer_type();
2059 
2060  switch (layerType)
2061  {
2062  case Layer::Layer_ConcatLayer:
2063  return graph->layers()->Get(layerIndex)->layer_as_ConcatLayer()->descriptor();
2064  case Layer::Layer_MergerLayer:
2065  return graph->layers()->Get(layerIndex)->layer_as_MergerLayer()->descriptor();
2066  default:
2067  throw armnn::Exception("unknown layer type, should be concat or merger");
2068  }
2069 }

◆ ToActivationFunction()

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

Definition at line 482 of file Deserializer.cpp.

483 {
484  switch (function)
485  {
486  case armnnSerializer::ActivationFunction_Sigmoid:
488  case armnnSerializer::ActivationFunction_TanH:
490  case armnnSerializer::ActivationFunction_Linear:
492  case armnnSerializer::ActivationFunction_ReLu:
494  case armnnSerializer::ActivationFunction_BoundedReLu:
496  case armnnSerializer::ActivationFunction_LeakyReLu:
498  case armnnSerializer::ActivationFunction_Abs:
500  case armnnSerializer::ActivationFunction_Sqrt:
502  case armnnSerializer::ActivationFunction_Square:
504  case armnnSerializer::ActivationFunction_Elu:
506  case armnnSerializer::ActivationFunction_HardSwish:
508  case armnnSerializer::ActivationFunction_Gelu:
510  default:
512  }
513 }

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

◆ ToArgMinMaxFunction()

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

Definition at line 515 of file Deserializer.cpp.

516 {
517  switch (function)
518  {
519  case armnnSerializer::ArgMinMaxFunction::ArgMinMaxFunction_Max:
521  case armnnSerializer::ArgMinMaxFunction::ArgMinMaxFunction_Min:
522  default:
524  }
525 }

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

◆ ToComparisonOperation()

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

Definition at line 546 of file Deserializer.cpp.

547 {
548  switch (operation)
549  {
550  case armnnSerializer::ComparisonOperation::ComparisonOperation_Equal:
552  case armnnSerializer::ComparisonOperation::ComparisonOperation_Greater:
554  case armnnSerializer::ComparisonOperation::ComparisonOperation_GreaterOrEqual:
556  case armnnSerializer::ComparisonOperation::ComparisonOperation_Less:
558  case armnnSerializer::ComparisonOperation::ComparisonOperation_LessOrEqual:
560  case armnnSerializer::ComparisonOperation::ComparisonOperation_NotEqual:
561  default:
563  }
564 }

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

◆ ToConstTensor()

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

Definition at line 782 of file Deserializer.cpp.

783 {
784  CHECK_CONST_TENSOR_PTR(constTensorPtr);
785  armnn::TensorInfo tensorInfo = ToTensorInfo(constTensorPtr->info());
786  tensorInfo.SetConstant();
787 
788  switch (constTensorPtr->data_type())
789  {
790  case ConstTensorData_ByteData:
791  {
792  auto byteData = constTensorPtr->data_as_ByteData()->data();
793  CHECK_CONST_TENSOR_SIZE(byteData->size(), tensorInfo.GetNumElements());
794  return armnn::ConstTensor(tensorInfo, byteData->data());
795  }
796  case ConstTensorData_ShortData:
797  {
798  auto shortData = constTensorPtr->data_as_ShortData()->data();
799  CHECK_CONST_TENSOR_SIZE(shortData->size(), tensorInfo.GetNumElements());
800  return armnn::ConstTensor(tensorInfo, shortData->data());
801  }
802  case ConstTensorData_IntData:
803  {
804  auto intData = constTensorPtr->data_as_IntData()->data();
805  CHECK_CONST_TENSOR_SIZE(intData->size(), tensorInfo.GetNumElements());
806  return armnn::ConstTensor(tensorInfo, intData->data());
807  }
808  case ConstTensorData_LongData:
809  {
810  auto longData = constTensorPtr->data_as_LongData()->data();
811  CHECK_CONST_TENSOR_SIZE(longData->size(), tensorInfo.GetNumElements());
812  return armnn::ConstTensor(tensorInfo, longData->data());
813  }
814  default:
815  {
816  CheckLocation location = CHECK_LOCATION();
817  throw ParseException(fmt::format("Unsupported data type {0} = {1}. {2}",
818  constTensorPtr->data_type(),
819  EnumNameConstTensorData(constTensorPtr->data_type()),
820  location.AsString()));
821  }
822  }
823 }

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

◆ ToDataLayout()

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

Definition at line 466 of file Deserializer.cpp.

467 {
468  switch (dataLayout)
469  {
470  case armnnSerializer::DataLayout::DataLayout_NHWC:
472  case armnnSerializer::DataLayout::DataLayout_NDHWC:
474  case armnnSerializer::DataLayout::DataLayout_NCDHW:
476  case armnnSerializer::DataLayout::DataLayout_NCHW:
477  default:
479  }
480 }

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

◆ ToElementwiseBinaryOperation()

armnn::BinaryOperation armnnDeserializer::ToElementwiseBinaryOperation ( armnnSerializer::BinaryOperation  operation)

Definition at line 598 of file Deserializer.cpp.

599 {
600  switch (operation)
601  {
602  case armnnSerializer::BinaryOperation::BinaryOperation_Add:
604  case armnnSerializer::BinaryOperation::BinaryOperation_Div:
606  case armnnSerializer::BinaryOperation::BinaryOperation_Maximum:
608  case armnnSerializer::BinaryOperation::BinaryOperation_Minimum:
610  case armnnSerializer::BinaryOperation::BinaryOperation_Mul:
612  case armnnSerializer::BinaryOperation::BinaryOperation_Sub:
614  case armnnSerializer::BinaryOperation::BinaryOperation_SqDiff:
616  case armnnSerializer::BinaryOperation::BinaryOperation_Power:
618  default:
619  throw armnn::InvalidArgumentException("Binary operation unknown");
620  }
621 }

References armnn::Add, armnn::Div, armnn::Maximum, armnn::Minimum, armnn::Mul, armnn::Power, armnn::SqDiff, and armnn::Sub.

◆ ToElementwiseUnaryOperation()

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

Definition at line 623 of file Deserializer.cpp.

624 {
625  switch (operation)
626  {
627  case armnnSerializer::UnaryOperation::UnaryOperation_Abs:
629  case armnnSerializer::UnaryOperation::UnaryOperation_Ceil:
631  case armnnSerializer::UnaryOperation::UnaryOperation_Rsqrt:
633  case armnnSerializer::UnaryOperation::UnaryOperation_Sqrt:
635  case armnnSerializer::UnaryOperation::UnaryOperation_Exp:
637  case armnnSerializer::UnaryOperation::UnaryOperation_Neg:
639  case armnnSerializer::UnaryOperation::UnaryOperation_LogicalNot:
641  case armnnSerializer::UnaryOperation::UnaryOperation_Log:
643  case armnnSerializer::UnaryOperation::UnaryOperation_Sin:
645  default:
646  throw armnn::InvalidArgumentException("Unary operation unknown");
647  }
648 }

References armnn::Abs, armnn::Ceil, armnn::Exp, armnn::Log, armnn::LogicalNot, armnn::Neg, armnn::Rsqrt, armnn::Sin, and armnn::Sqrt.

◆ ToLogicalBinaryOperation()

armnn::LogicalBinaryOperation armnnDeserializer::ToLogicalBinaryOperation ( armnnSerializer::LogicalBinaryOperation  operation)

Definition at line 585 of file Deserializer.cpp.

586 {
587  switch (operation)
588  {
589  case armnnSerializer::LogicalBinaryOperation::LogicalBinaryOperation_LogicalAnd:
591  case armnnSerializer::LogicalBinaryOperation::LogicalBinaryOperation_LogicalOr:
593  default:
594  throw armnn::InvalidArgumentException("Logical Binary operation unknown");
595  }
596 }

References armnn::LogicalAnd, and armnn::LogicalOr.

◆ ToPaddingMode()

armnn::PaddingMode armnnDeserializer::ToPaddingMode ( armnnSerializer::PaddingMode  paddingMode)

Definition at line 650 of file Deserializer.cpp.

651 {
652  switch (paddingMode)
653  {
654  case armnnSerializer::PaddingMode::PaddingMode_Reflect:
656  case armnnSerializer::PaddingMode::PaddingMode_Symmetric:
658  default:
660  }
661 }

References armnn::Constant, armnn::Reflect, and armnn::Symmetric.

◆ ToReduceOperation()

armnn::ReduceOperation armnnDeserializer::ToReduceOperation ( armnnSerializer::ReduceOperation  operation)

Definition at line 566 of file Deserializer.cpp.

567 {
568  switch (operation)
569  {
570  case armnnSerializer::ReduceOperation::ReduceOperation_Sum:
572  case armnnSerializer::ReduceOperation::ReduceOperation_Max:
574  case armnnSerializer::ReduceOperation::ReduceOperation_Mean:
576  case armnnSerializer::ReduceOperation::ReduceOperation_Min:
578  case armnnSerializer::ReduceOperation::ReduceOperation_Prod:
580  default:
582  }
583 }

References armnn::Max, armnn::Mean, armnn::Min, armnn::Prod, and armnn::Sum.

◆ ToResizeMethod()

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

Definition at line 663 of file Deserializer.cpp.

664 {
665  switch (method)
666  {
667  case armnnSerializer::ResizeMethod_NearestNeighbor:
669  case armnnSerializer::ResizeMethod_Bilinear:
671  default:
673  }
674 }

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

◆ ToScatterNdFunction()

armnn::ScatterNdFunction armnnDeserializer::ToScatterNdFunction ( armnnSerializer::ScatterNdFunction  function)

Definition at line 527 of file Deserializer.cpp.

528 {
529  switch (function)
530  {
531  case armnnSerializer::ScatterNdFunction_Update:
533  case armnnSerializer::ScatterNdFunction_Add:
535  case armnnSerializer::ScatterNdFunction_Sub:
537  case armnnSerializer::ScatterNdFunction_Max:
539  case armnnSerializer::ScatterNdFunction_Min:
541  default:
543  }
544 }

References armnn::Add, armnn::Max, armnn::Min, armnn::Sub, and armnn::Update.

◆ ToTensorInfo()

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

Definition at line 676 of file Deserializer.cpp.

677 {
678  armnn::DataType type;
679  CHECK_TENSOR_PTR(tensorPtr);
680 
681  switch (tensorPtr->dataType())
682  {
683  case DataType_QAsymmS8:
685  break;
686  case DataType_QSymmS8:
688  break;
689  case DataType_QuantisedAsymm8:
690  case DataType_QAsymmU8:
692  break;
693  case DataType_QSymmS16:
694  case DataType_QuantisedSymm16:
696  break;
697  case DataType_Signed32:
699  break;
700  case DataType_Signed64:
702  break;
703  case DataType_Float32:
705  break;
706  case DataType_Float16:
708  break;
709  case DataType_Boolean:
711  break;
712  default:
713  {
714  CheckLocation location = CHECK_LOCATION();
715  throw ParseException(fmt::format("Unsupported data type {0} = {1}. {2}",
716  tensorPtr->dataType(),
717  EnumNameDataType(tensorPtr->dataType()),
718  location.AsString()));
719  }
720  }
721 
722  float quantizationScale = tensorPtr->quantizationScale();
723  int32_t quantizationOffset = tensorPtr->quantizationOffset();
724 
725  if (tensorPtr->dimensionality() == static_cast<unsigned int>(Dimensionality::Scalar))
726  {
728  type,
729  quantizationScale,
730  quantizationOffset);
731  }
732  else if (tensorPtr->dimensionality() == static_cast<unsigned int>(Dimensionality::NotSpecified))
733  {
734  armnn::TensorInfo result(TensorShape{Dimensionality::NotSpecified},
735  type,
736  quantizationScale,
737  quantizationOffset);
738  return result;
739  }
740 
741  auto dimensions = tensorPtr->dimensions();
742  unsigned int size = dimensions->size();
743  std::vector<unsigned int> outputDims(dimensions->begin(), dimensions->begin() + size);
744  bool dimensionsSpecificity[armnn::MaxNumOfTensorDimensions];
745  std::fill_n(dimensionsSpecificity, armnn::MaxNumOfTensorDimensions, true);
746  // For backwards compatibility check if the dimensionSpecificity vector is present first.
747  // The default is to have dimensionSpecificity set to all true's anyway.
748  if (tensorPtr->dimensionSpecificity() != nullptr)
749  {
750  auto dimensionSpecificity = tensorPtr->dimensionSpecificity();
751  size = dimensionSpecificity->size();
752  for (unsigned int i = 0; i < size; ++i)
753  {
754  dimensionsSpecificity[i] = dimensionSpecificity->Get(i);
755  }
756  }
757  // Construct a TensorShape
758  TensorShape shape(size, outputDims.data(), dimensionsSpecificity);
759 
760  auto quantizationScales = tensorPtr->quantizationScales();
761  if (quantizationScales)
762  {
763  unsigned int quantizationScalesSize = quantizationScales->size();
764  std::vector<float> scales(quantizationScales->begin(), quantizationScales->begin() + quantizationScalesSize);
765  unsigned int quantizationDim = tensorPtr->quantizationDim();
766  armnn::TensorInfo result(shape,
767  type,
768  scales,
769  quantizationDim);
770  return result;
771  }
772 
773  // two statements (on purpose) for easier debugging:
774  armnn::TensorInfo result(shape,
775  type,
776  quantizationScale,
777  quantizationOffset);
778 
779  return result;
780 }

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

Referenced by TfLiteParserImpl::GetNetworkInputBindingInfo(), TfLiteParserImpl::GetNetworkOutputBindingInfo(), and ToConstTensor().

armnn::ArgMinMaxFunction::Max
@ Max
armnn::PaddingMode::Symmetric
@ Symmetric
armnn::BinaryOperation::Mul
@ Mul
armnn::DataType::Boolean
@ Boolean
armnn::TensorInfo::GetNumElements
unsigned int GetNumElements() const
Definition: Tensor.hpp:198
armnn::BinaryOperation::Add
@ Add
armnn::ComparisonOperation::LessOrEqual
@ LessOrEqual
armnn::ScatterNdFunction::Min
@ Min
armnn::DataLayout::NCDHW
@ NCDHW
armnn::ActivationFunction::LeakyReLu
@ LeakyReLu
armnn::ScatterNdFunction::Sub
@ Sub
armnn::DataLayout::NHWC
@ NHWC
armnn::ActivationFunction::Sqrt
@ Sqrt
armnn::TensorInfo
Definition: Tensor.hpp:152
armnn::BinaryOperation::Sub
@ Sub
CHECK_LOCATION
#define CHECK_LOCATION()
Definition: Exceptions.hpp:203
armnn::DataType::Float32
@ Float32
armnn::ActivationFunction::TanH
@ TanH
armnn::LogicalBinaryOperation::LogicalOr
@ LogicalOr
CHECK_CONST_TENSOR_PTR
#define CHECK_CONST_TENSOR_PTR(TENSOR_PTR)
Definition: Deserializer.cpp:178
armnn::MaxNumOfTensorDimensions
constexpr unsigned int MaxNumOfTensorDimensions
Definition: Types.hpp:31
armnn::DataType::QAsymmU8
@ QAsymmU8
armnn::ActivationFunction::BoundedReLu
@ BoundedReLu
min(a, max(b, input)) ReLu1 & ReLu6.
armnn::DataType::QSymmS8
@ QSymmS8
armnn::UnaryOperation::Neg
@ Neg
armnn::CheckLocation::AsString
std::string AsString() const
Definition: Exceptions.hpp:29
armnn::ReduceOperation::Mean
@ Mean
armnn::ActivationFunction::HardSwish
@ HardSwish
armnn::DataType::QSymmS16
@ QSymmS16
armnn::ActivationFunction::Gelu
@ Gelu
armnn::ComparisonOperation::NotEqual
@ NotEqual
armnn::ComparisonOperation::GreaterOrEqual
@ GreaterOrEqual
armnn::DataLayout::NDHWC
@ NDHWC
armnn::TensorShape
Definition: Tensor.hpp:20
armnn::BinaryOperation::Maximum
@ Maximum
armnn::DataType::Float16
@ Float16
armnn::CheckLocation
Definition: Exceptions.hpp:14
armnn::BinaryOperation::SqDiff
@ SqDiff
armnn::UnaryOperation::Rsqrt
@ Rsqrt
armnn::TensorShape::GetNumDimensions
unsigned int GetNumDimensions() const
Function that returns the tensor rank.
Definition: Tensor.cpp:174
armnn::ComparisonOperation::Less
@ Less
armnn::UnaryOperation::Sqrt
@ Sqrt
armnn::UnaryOperation::LogicalNot
@ LogicalNot
armnn::UnaryOperation::Exp
@ Exp
armnn::DataType
DataType
Definition: Types.hpp:48
armnn::Dimensionality::Scalar
@ Scalar
armnn::ActivationFunction::Elu
@ Elu
armnn::InvalidArgumentException
Definition: Exceptions.hpp:80
armnn::UnaryOperation::Sin
@ Sin
armnn::ActivationFunction::Linear
@ Linear
CHECK_TENSOR_PTR
#define CHECK_TENSOR_PTR(TENSOR_PTR)
Definition: Deserializer.cpp:172
armnn::ReduceOperation::Sum
@ Sum
armnn::Exception
Base class for all ArmNN exceptions so that users can filter to just those.
Definition: Exceptions.hpp:46
armnn::ScatterNdFunction::Add
@ Add
armnn::ResizeMethod::NearestNeighbor
@ NearestNeighbor
armnn::BinaryOperation::Power
@ Power
CHECK_CONST_TENSOR_SIZE
#define CHECK_CONST_TENSOR_SIZE(CONST_TENSOR_SIZE, TENSOR_SIZE)
Definition: Deserializer.cpp:175
armnn::DataType::Signed32
@ Signed32
armnn::UnaryOperation::Ceil
@ Ceil
armnn::ReduceOperation::Prod
@ Prod
armnn::ActivationFunction::Abs
@ Abs
armnn::DataType::QAsymmS8
@ QAsymmS8
armnn::ResizeMethod::Bilinear
@ Bilinear
armnn::ArgMinMaxFunction::Min
@ Min
armnn::UnaryOperation::Log
@ Log
armnn::LogicalBinaryOperation::LogicalAnd
@ LogicalAnd
armnnDeserializer::ToTensorInfo
armnn::TensorInfo ToTensorInfo(TensorRawPtr tensorPtr)
Definition: Deserializer.cpp:676
armnn::ParseException
Definition: Exceptions.hpp:92
armnn::ScatterNdFunction::Update
@ Update
armnn::BinaryOperation::Minimum
@ Minimum
armnn::ScatterNdFunction::Max
@ Max
armnn::ActivationFunction::ReLu
@ ReLu
armnn::UnaryOperation::Abs
@ Abs
armnn::ReduceOperation::Min
@ Min
armnn::ConstTensor
A tensor defined by a TensorInfo (shape and data type) and an immutable backing store.
Definition: Tensor.hpp:329
armnn::ActivationFunction::Square
@ Square
armnn::TensorInfo::SetConstant
void SetConstant(const bool IsConstant=true)
Marks the data corresponding to this tensor info as constant.
Definition: Tensor.cpp:518
armnn::BinaryOperation::Div
@ Div
armnn::DataType::Signed64
@ Signed64
armnn::ComparisonOperation::Equal
@ Equal
armnn::PaddingMode::Reflect
@ Reflect
armnn::PaddingMode::Constant
@ Constant
armnn::ReduceOperation::Max
@ Max
armnn::DataLayout::NCHW
@ NCHW
armnn::ActivationFunction::Sigmoid
@ Sigmoid
armnn::ComparisonOperation::Greater
@ Greater