armnnTfLiteParser Namespace Reference

Classes
class	ITfLiteParser
class	TfLiteParser

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 23 of file ITfLiteParser.hpp.

Function Documentation

ComputeWrappedIndex()

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

Definition at line 2594 of file TfLiteParser.cpp.

References ARMNN_ASSERT, CHECK_LOCATION, CHECK_MODEL, CHECK_VALID_SIZE, CHECKED_NON_NEGATIVE, IOutputSlot::Connect(), TfLiteParser::GetBuffer(), TfLiteParser::GetInputs(), IConnectableLayer::GetInputSlot(), TfLiteParser::GetInputTensorIds(), IConnectableLayer::GetName(), TensorInfo::GetNumBytes(), TensorInfo::GetNumDimensions(), TensorInfo::GetNumElements(), IConnectableLayer::GetNumOutputSlots(), TfLiteParser::GetOutputs(), IConnectableLayer::GetOutputSlot(), TfLiteParser::GetOutputTensorIds(), TensorInfo::GetShape(), ActivationDescriptor::m_A, ActivationDescriptor::m_B, ActivationDescriptor::m_Function, armnn::MaxNumOfTensorDimensions, armnn::numeric_cast(), options, IOutputSlot::SetTensorInfo(), 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);
}