armnnTfLiteParser Namespace Reference

Classes
class	ITfLiteParser
class	TfLiteParserImpl

Typedefs
using	BindingPointInfo = armnn::BindingPointInfo
using	ITfLiteParserPtr = std::unique_ptr< ITfLiteParser, void(*)(ITfLiteParser *parser)>

Functions
unsigned int	ComputeWrappedIndex (int idx, unsigned int numDimsIn)

Typedef Documentation

BindingPointInfo

using BindingPointInfo = armnn::BindingPointInfo

Definition at line 20 of file ITfLiteParser.hpp.

ITfLiteParserPtr

using ITfLiteParserPtr = std::unique_ptr<ITfLiteParser, void(*)(ITfLiteParser* parser)>

Definition at line 24 of file ITfLiteParser.hpp.

Function Documentation

ComputeWrappedIndex()

unsigned int armnnTfLiteParser::ComputeWrappedIndex	(	int	idx,
		unsigned int	numDimsIn
	)

Definition at line 2983 of file TfLiteParser.cpp.

References armnn::Abs, ARMNN_ASSERT, CHECK_LOCATION, CHECK_MODEL, CHECK_VALID_SIZE, CHECKED_NON_NEGATIVE, IOutputSlot::Connect(), armnn::Equal, armnn::Exp, TfLiteParserImpl::GetBuffer(), armnn::GetComparisonOperationAsCString(), TensorInfo::GetDataType(), TfLiteParserImpl::GetInputs(), IConnectableLayer::GetInputSlot(), TfLiteParserImpl::GetInputTensorIds(), IConnectableLayer::GetName(), TensorInfo::GetNumBytes(), TensorInfo::GetNumDimensions(), TensorInfo::GetNumElements(), IConnectableLayer::GetNumOutputSlots(), TfLiteParserImpl::GetOutputs(), IConnectableLayer::GetOutputSlot(), TfLiteParserImpl::GetOutputTensorIds(), TensorInfo::GetShape(), armnn::GetUnaryOperationAsCString(), armnn::Greater, armnn::GreaterOrEqual, armnn::Less, armnn::LessOrEqual, armnn::LogicalNot, ActivationDescriptor::m_A, ArgMinMaxDescriptor::m_Axis, GatherDescriptor::m_Axis, ActivationDescriptor::m_B, ActivationDescriptor::m_Function, ArgMinMaxDescriptor::m_Function, ReduceDescriptor::m_KeepDims, ComparisonDescriptor::m_Operation, ElementwiseUnaryDescriptor::m_Operation, ReduceDescriptor::m_ReduceOperation, ReduceDescriptor::m_vAxis, armnn::Max, armnn::MaxNumOfTensorDimensions, armnn::Min, armnn::Neg, armnn::NotEqual, armnn::numeric_cast(), armnn::Rsqrt, IOutputSlot::SetTensorInfo(), armnn::Signed32, armnn::Signed64, armnn::Sum, and armnnDeserializer::ToTensorInfo().

{
    int numDims = armnn::numeric_cast<int>(numDimsIn);
    int v = idx < 0 ? numDims + idx : idx;
    ARMNN_ASSERT(v >= 0);
    ARMNN_ASSERT(v < numDims);

    return static_cast<unsigned int>(v);
}