NetworkImpl Class Reference

Private implementation of INetwork. More...

#include <Network.hpp>

Public Member Functions
	NetworkImpl (NetworkOptions networkOptions={})
	~NetworkImpl ()
const Graph &	GetGraph () const
Status	PrintGraph ()
IConnectableLayer *	AddInputLayer (LayerBindingId id, const char *name=nullptr)
IConnectableLayer *	AddArgMinMaxLayer (const ArgMinMaxDescriptor &desc, const char *name=nullptr)
IConnectableLayer *	AddBatchToSpaceNdLayer (const BatchToSpaceNdDescriptor &batchToSpaceNdDescriptor, const char *name=nullptr)
IConnectableLayer *	AddComparisonLayer (const ComparisonDescriptor &comparisonDescriptor, const char *name=nullptr)
IConnectableLayer *	AddConcatLayer (const ConcatDescriptor &concatDescriptor, const char *name=nullptr)
IConnectableLayer *	AddConvolution2dLayer (const Convolution2dDescriptor &convolution2dDescriptor, const ConstTensor &weights, const Optional< ConstTensor > &biases, const char *name=nullptr)
IConnectableLayer *	AddConvolution2dLayer (const Convolution2dDescriptor &convolution2dDescriptor, const ConstTensor &weights, const char *name=nullptr)
IConnectableLayer *	AddConvolution2dLayer (const Convolution2dDescriptor &convolution2dDescriptor, const ConstTensor &weights, const ConstTensor &biases, const char *name=nullptr)
IConnectableLayer *	AddDepthToSpaceLayer (const DepthToSpaceDescriptor &depthToSpaceDescriptor, const char *name=nullptr)
IConnectableLayer *	AddDepthwiseConvolution2dLayer (const DepthwiseConvolution2dDescriptor &convolution2dDescriptor, const ConstTensor &weights, const Optional< ConstTensor > &biases, const char *name=nullptr)
IConnectableLayer *	AddDepthwiseConvolution2dLayer (const DepthwiseConvolution2dDescriptor &convolution2dDescriptor, const ConstTensor &weights, const char *name=nullptr)
IConnectableLayer *	AddDepthwiseConvolution2dLayer (const DepthwiseConvolution2dDescriptor &convolution2dDescriptor, const ConstTensor &weights, const ConstTensor &biases, const char *name=nullptr)
IConnectableLayer *	AddDequantizeLayer (const char *name=nullptr)
IConnectableLayer *	AddDetectionPostProcessLayer (const DetectionPostProcessDescriptor &descriptor, const ConstTensor &anchors, const char *name=nullptr)
IConnectableLayer *	AddElementwiseUnaryLayer (const ElementwiseUnaryDescriptor &elementwiseUnaryDescriptor, const char *name=nullptr)
IConnectableLayer *	AddFillLayer (const FillDescriptor &fillDescriptor, const char *name=nullptr)
IConnectableLayer *	AddFullyConnectedLayer (const FullyConnectedDescriptor &fullyConnectedDescriptor, const ConstTensor &weights, const Optional< ConstTensor > &biases, const char *name=nullptr)
IConnectableLayer *	AddFullyConnectedLayer (const FullyConnectedDescriptor &fullyConnectedDescriptor, const ConstTensor &weights, const char *name=nullptr)
IConnectableLayer *	AddFullyConnectedLayer (const FullyConnectedDescriptor &fullyConnectedDescriptor, const ConstTensor &weights, const ConstTensor &biases, const char *name=nullptr)
IConnectableLayer *	AddGatherLayer (const char *name=nullptr)
IConnectableLayer *	AddGatherLayer (const GatherDescriptor &gatherDescriptor, const char *name=nullptr)
IConnectableLayer *	AddPermuteLayer (const PermuteDescriptor &permuteDescriptor, const char *name=nullptr)
IConnectableLayer *	AddPooling2dLayer (const Pooling2dDescriptor &pooling2dDescriptor, const char *name=nullptr)
IConnectableLayer *	AddActivationLayer (const ActivationDescriptor &activationDescriptor, const char *name=nullptr)
IConnectableLayer *	AddNormalizationLayer (const NormalizationDescriptor &normalizationDescriptor, const char *name=nullptr)
IConnectableLayer *	AddSliceLayer (const SliceDescriptor &sliceDescriptor, const char *name=nullptr)
IConnectableLayer *	AddSoftmaxLayer (const SoftmaxDescriptor &softmaxDescriptor, const char *name=nullptr)
IConnectableLayer *	AddSplitterLayer (const ViewsDescriptor &splitterDescriptor, const char *name=nullptr)
IConnectableLayer *	AddMergerLayer (const MergerDescriptor &mergerDescriptor, const char *name=nullptr)
IConnectableLayer *	AddAbsLayer (const char *name=nullptr)
IConnectableLayer *	AddAdditionLayer (const char *name=nullptr)
IConnectableLayer *	AddMultiplicationLayer (const char *name=nullptr)
IConnectableLayer *	AddBatchNormalizationLayer (const BatchNormalizationDescriptor &desc, const ConstTensor &mean, const ConstTensor &variance, const ConstTensor &beta, const ConstTensor &gamma, const char *name=nullptr)
IConnectableLayer *	AddRankLayer (const char *name=nullptr)
IConnectableLayer *	AddResizeBilinearLayer (const ResizeBilinearDescriptor &resizeDesc, const char *name=nullptr)
IConnectableLayer *	AddResizeLayer (const ResizeDescriptor &resizeDescriptor, const char *name=nullptr)
IConnectableLayer *	AddReduceLayer (const ReduceDescriptor &reduceDescriptor, const char *name=nullptr)
IConnectableLayer *	AddInstanceNormalizationLayer (const InstanceNormalizationDescriptor &desc, const char *name=nullptr)
IConnectableLayer *	AddL2NormalizationLayer (const L2NormalizationDescriptor &desc, const char *name=nullptr)
IConnectableLayer *	AddLogSoftmaxLayer (const LogSoftmaxDescriptor &logSoftmaxDescriptor, const char *name=nullptr)
IConnectableLayer *	AddConstantLayer (const ConstTensor &input, const char *name=nullptr)
IConnectableLayer *	AddReshapeLayer (const ReshapeDescriptor &reshapeDescriptor, const char *name=nullptr)
IConnectableLayer *	AddSpaceToBatchNdLayer (const SpaceToBatchNdDescriptor &spaceToBatchNdDescriptor, const char *name=nullptr)
IConnectableLayer *	AddSpaceToDepthLayer (const SpaceToDepthDescriptor &spaceToDepthDescriptor, const char *name=nullptr)
IConnectableLayer *	AddFloorLayer (const char *name=nullptr)
IConnectableLayer *	AddOutputLayer (LayerBindingId id, const char *name=nullptr)
IConnectableLayer *	AddLstmLayer (const LstmDescriptor &descriptor, const LstmInputParams &params, const char *name=nullptr)
IConnectableLayer *	AddDivisionLayer (const char *name=nullptr)
IConnectableLayer *	AddSubtractionLayer (const char *name=nullptr)
IConnectableLayer *	AddMaximumLayer (const char *name=nullptr)
IConnectableLayer *	AddMeanLayer (const MeanDescriptor &meanDescriptor, const char *name=nullptr)
IConnectableLayer *	AddPadLayer (const PadDescriptor &padDescriptor, const char *name=nullptr)
IConnectableLayer *	AddQuantizeLayer (const char *name=nullptr)
IConnectableLayer *	AddStridedSliceLayer (const StridedSliceDescriptor &stridedSliceDescriptor, const char *name=nullptr)
IConnectableLayer *	AddMinimumLayer (const char *name=nullptr)
IConnectableLayer *	AddGreaterLayer (const char *name=nullptr)
IConnectableLayer *	AddEqualLayer (const char *name=nullptr)
IConnectableLayer *	AddRsqrtLayer (const char *name=nullptr)
IConnectableLayer *	AddMergeLayer (const char *name=nullptr)
IConnectableLayer *	AddSwitchLayer (const char *name=nullptr)
IConnectableLayer *	AddPreluLayer (const char *name=nullptr)
IConnectableLayer *	AddTransposeConvolution2dLayer (const TransposeConvolution2dDescriptor &descriptor, const ConstTensor &weights, const Optional< ConstTensor > &biases, const char *name=nullptr)
IConnectableLayer *	AddTransposeLayer (const TransposeDescriptor &transposeDescriptor, const char *name=nullptr)
IConnectableLayer *	AddStackLayer (const StackDescriptor &stackDescriptor, const char *name=nullptr)
IConnectableLayer *	AddStandInLayer (const StandInDescriptor &descriptor, const char *name=nullptr)
IConnectableLayer *	AddQLstmLayer (const QLstmDescriptor &descriptor, const LstmInputParams &params, const char *name=nullptr)
IConnectableLayer *	AddQuantizedLstmLayer (const QuantizedLstmInputParams &params, const char *name=nullptr)
IConnectableLayer *	AddLogicalBinaryLayer (const LogicalBinaryDescriptor &logicalBinaryDescriptor, const char *name=nullptr)
void	Accept (ILayerVisitor &visitor) const
void	ExecuteStrategy (IStrategy &strategy) const

Detailed Description

Private implementation of INetwork.

Definition at line 31 of file Network.hpp.

Constructor & Destructor Documentation

NetworkImpl()

NetworkImpl

(

NetworkOptions

networkOptions = {}

)

Definition at line 1667 of file Network.cpp.

: m_NetworkOptions(networkOptions),
  m_Graph(std::make_unique<Graph>(GetShapeInferenceMethod()))
{}

~NetworkImpl()

~NetworkImpl

(

)

Definition at line 1672 of file Network.cpp.

{
}

Member Function Documentation

Accept()

void Accept

(

ILayerVisitor &

visitor

)

const

Definition at line 2505 of file Network.cpp.

References NetworkImpl::GetGraph().

Referenced by NetworkImpl::GetGraph().

{
    for (auto layer : GetGraph())
    {
        layer->Accept(visitor);
    };
}

AddAbsLayer()

IConnectableLayer * AddAbsLayer

(

const char *

name = nullptr

)

Definition at line 1942 of file Network.cpp.

References armnn::Abs, and NetworkImpl::AddElementwiseUnaryLayer().

Referenced by NetworkImpl::GetGraph().

{
    return AddElementwiseUnaryLayer(ElementwiseUnaryDescriptor(UnaryOperation::Abs), name);
}

AddActivationLayer()

IConnectableLayer * AddActivationLayer	(	const ActivationDescriptor &	activationDescriptor,
		const char *	name = nullptr
	)

Definition at line 1890 of file Network.cpp.

Referenced by BOOST_AUTO_TEST_CASE(), and NetworkImpl::GetGraph().

{
    return m_Graph->AddLayer<ActivationLayer>(activationDescriptor, name);
}

AddAdditionLayer()

IConnectableLayer * AddAdditionLayer

(

const char *

name = nullptr

)

Definition at line 1947 of file Network.cpp.

Referenced by BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), and NetworkImpl::GetGraph().

{
    return m_Graph->AddLayer<AdditionLayer>(name);
}

AddArgMinMaxLayer()

IConnectableLayer * AddArgMinMaxLayer	(	const ArgMinMaxDescriptor &	desc,
		const char *	name = nullptr
	)

Definition at line 1896 of file Network.cpp.

Referenced by NetworkImpl::GetGraph().

{
    return m_Graph->AddLayer<ArgMinMaxLayer>(argMinMaxDescriptor, name);
}

AddBatchNormalizationLayer()

IConnectableLayer * AddBatchNormalizationLayer	(	const BatchNormalizationDescriptor &	desc,
		const ConstTensor &	mean,
		const ConstTensor &	variance,
		const ConstTensor &	beta,
		const ConstTensor &	gamma,
		const char *	name = nullptr
	)

Definition at line 1962 of file Network.cpp.

References BatchNormalizationLayer::m_Mean.

Referenced by BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), and NetworkImpl::GetGraph().

{
    const auto layer = m_Graph->AddLayer<BatchNormalizationLayer>(desc, name);

    layer->m_Mean = std::make_unique<ScopedCpuTensorHandle>(mean);
    layer->m_Variance = std::make_unique<ScopedCpuTensorHandle>(variance);
    layer->m_Beta = std::make_unique<ScopedCpuTensorHandle>(beta);
    layer->m_Gamma = std::make_unique<ScopedCpuTensorHandle>(gamma);

    return layer;
}

AddBatchToSpaceNdLayer()

IConnectableLayer * AddBatchToSpaceNdLayer	(	const BatchToSpaceNdDescriptor &	batchToSpaceNdDescriptor,
		const char *	name = nullptr
	)

Definition at line 1687 of file Network.cpp.

Referenced by NetworkImpl::GetGraph().

{
    return m_Graph->AddLayer<BatchToSpaceNdLayer>(batchToSpaceNdDescriptor, name);
}

AddComparisonLayer()

IConnectableLayer * AddComparisonLayer	(	const ComparisonDescriptor &	comparisonDescriptor,
		const char *	name = nullptr
	)

Definition at line 1693 of file Network.cpp.

Referenced by NetworkImpl::AddEqualLayer(), NetworkImpl::AddGreaterLayer(), and NetworkImpl::GetGraph().

{
    return m_Graph->AddLayer<ComparisonLayer>(comparisonDescriptor, name);
}

AddConcatLayer()

IConnectableLayer * AddConcatLayer	(	const ConcatDescriptor &	concatDescriptor,
		const char *	name = nullptr
	)

Definition at line 1758 of file Network.cpp.

References OptionalBase::has_value(), Convolution2dDescriptor::m_BiasEnabled, Convolution2dLayer::m_Weight, and OptionalReferenceSwitch< std::is_reference< T >::value, T >::value().

Referenced by NetworkImpl::AddMergerLayer(), BOOST_AUTO_TEST_CASE(), and NetworkImpl::GetGraph().

{
    return m_Graph->AddLayer<ConcatLayer>(concatDescriptor, name);
}

AddConstantLayer()

IConnectableLayer * AddConstantLayer	(	const ConstTensor &	input,
		const char *	name = nullptr
	)

Definition at line 2027 of file Network.cpp.

References ConstantLayer::m_LayerOutput.

Referenced by armnn::BOOST_AUTO_TEST_CASE(), and NetworkImpl::GetGraph().

{
    auto layer = m_Graph->AddLayer<ConstantLayer>(name);

    layer->m_LayerOutput = std::make_unique<ScopedCpuTensorHandle>(input);

    return layer;
}

AddConvolution2dLayer() [1/3]

IConnectableLayer * AddConvolution2dLayer	(	const Convolution2dDescriptor &	convolution2dDescriptor,
		const ConstTensor &	weights,
		const Optional< ConstTensor > &	biases,
		const char *	name = nullptr
	)

Definition at line 1786 of file Network.cpp.

Referenced by BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), and NetworkImpl::GetGraph().

{
    return AddConvolution2dLayerImpl(convolution2dDescriptor, weights, biases, name);
}

AddConvolution2dLayer() [2/3]

IConnectableLayer * AddConvolution2dLayer	(	const Convolution2dDescriptor &	convolution2dDescriptor,
		const ConstTensor &	weights,
		const char *	name = nullptr
	)

Definition at line 1794 of file Network.cpp.

{
    Optional<ConstTensor> biases;
    return AddConvolution2dLayerImpl(convolution2dDescriptor, weights, biases, name);
}

AddConvolution2dLayer() [3/3]

IConnectableLayer * AddConvolution2dLayer	(	const Convolution2dDescriptor &	convolution2dDescriptor,
		const ConstTensor &	weights,
		const ConstTensor &	biases,
		const char *	name = nullptr
	)

Definition at line 1802 of file Network.cpp.

References OptionalBase::has_value(), DepthwiseConvolution2dDescriptor::m_BiasEnabled, DepthwiseConvolution2dLayer::m_Weight, and OptionalReferenceSwitch< std::is_reference< T >::value, T >::value().

{
    Optional<ConstTensor> optionalBiases(biases);
    return AddConvolution2dLayerImpl(convolution2dDescriptor, weights, optionalBiases, name);
}

AddDepthToSpaceLayer()

IConnectableLayer * AddDepthToSpaceLayer	(	const DepthToSpaceDescriptor &	depthToSpaceDescriptor,
		const char *	name = nullptr
	)

Definition at line 1834 of file Network.cpp.

Referenced by NetworkImpl::GetGraph().

{
    return m_Graph->AddLayer<DepthToSpaceLayer>(depthToSpaceDescriptor, name);
}

AddDepthwiseConvolution2dLayer() [1/3]

IConnectableLayer * AddDepthwiseConvolution2dLayer	(	const DepthwiseConvolution2dDescriptor &	convolution2dDescriptor,
		const ConstTensor &	weights,
		const Optional< ConstTensor > &	biases,
		const char *	name = nullptr
	)

Definition at line 1840 of file Network.cpp.

Referenced by armnn::BOOST_AUTO_TEST_CASE(), and NetworkImpl::GetGraph().

{
    return AddDepthwiseConvolution2dLayerImpl(convolution2dDescriptor, weights, biases, name);
}

AddDepthwiseConvolution2dLayer() [2/3]

IConnectableLayer * AddDepthwiseConvolution2dLayer	(	const DepthwiseConvolution2dDescriptor &	convolution2dDescriptor,
		const ConstTensor &	weights,
		const char *	name = nullptr
	)

Definition at line 1849 of file Network.cpp.

{
    Optional<ConstTensor> biases;
    return AddDepthwiseConvolution2dLayerImpl(convolution2dDescriptor, weights, biases, name);
}

AddDepthwiseConvolution2dLayer() [3/3]

IConnectableLayer * AddDepthwiseConvolution2dLayer	(	const DepthwiseConvolution2dDescriptor &	convolution2dDescriptor,
		const ConstTensor &	weights,
		const ConstTensor &	biases,
		const char *	name = nullptr
	)

Definition at line 1858 of file Network.cpp.

{
    Optional<ConstTensor> optionalBiases(biases);
    return AddDepthwiseConvolution2dLayerImpl(convolution2dDescriptor, weights, optionalBiases, name);
}

AddDequantizeLayer()

IConnectableLayer * AddDequantizeLayer

(

const char *

name = nullptr

)

Definition at line 2225 of file Network.cpp.

Referenced by NetworkImpl::GetGraph().

{
    return m_Graph->AddLayer<DequantizeLayer>(name);
}

AddDetectionPostProcessLayer()

IConnectableLayer * AddDetectionPostProcessLayer	(	const DetectionPostProcessDescriptor &	descriptor,
		const ConstTensor &	anchors,
		const char *	name = nullptr
	)

Definition at line 1868 of file Network.cpp.

References anchors(), and DetectionPostProcessLayer::m_Anchors.

Referenced by NetworkImpl::GetGraph().

{
    const auto layer = m_Graph->AddLayer<DetectionPostProcessLayer>(descriptor, name);

    layer->m_Anchors = std::make_unique<ScopedCpuTensorHandle>(anchors);

    return layer;
}

AddDivisionLayer()

IConnectableLayer * AddDivisionLayer

(

const char *

name = nullptr

)

Definition at line 2200 of file Network.cpp.

Referenced by NetworkImpl::GetGraph().

{
    return m_Graph->AddLayer<DivisionLayer>(name);
}

AddElementwiseUnaryLayer()

IConnectableLayer * AddElementwiseUnaryLayer	(	const ElementwiseUnaryDescriptor &	elementwiseUnaryDescriptor,
		const char *	name = nullptr
	)

Definition at line 1699 of file Network.cpp.

Referenced by NetworkImpl::AddAbsLayer(), NetworkImpl::AddRsqrtLayer(), and NetworkImpl::GetGraph().

{
    return m_Graph->AddLayer<ElementwiseUnaryLayer>(elementwiseUnaryDescriptor, name);
}

AddEqualLayer()

IConnectableLayer * AddEqualLayer

(

const char *

name = nullptr

)

Definition at line 2241 of file Network.cpp.

References NetworkImpl::AddComparisonLayer(), and armnn::Equal.

Referenced by NetworkImpl::GetGraph().

{
    return AddComparisonLayer(ComparisonDescriptor(ComparisonOperation::Equal), name);
}

AddFillLayer()

IConnectableLayer * AddFillLayer	(	const FillDescriptor &	fillDescriptor,
		const char *	name = nullptr
	)

Definition at line 1705 of file Network.cpp.

References OptionalBase::has_value(), FullyConnectedDescriptor::m_BiasEnabled, FullyConnectedLayer::m_Weight, and OptionalReferenceSwitch< std::is_reference< T >::value, T >::value().

Referenced by NetworkImpl::GetGraph().

{
    return m_Graph->AddLayer<FillLayer>(fillDescriptor, name);
}

AddFloorLayer()

IConnectableLayer * AddFloorLayer

(

const char *

name = nullptr

)

Definition at line 2054 of file Network.cpp.

Referenced by BOOST_AUTO_TEST_CASE(), and NetworkImpl::GetGraph().

{
    return m_Graph->AddLayer<FloorLayer>(name);
}

AddFullyConnectedLayer() [1/3]

IConnectableLayer * AddFullyConnectedLayer	(	const FullyConnectedDescriptor &	fullyConnectedDescriptor,
		const ConstTensor &	weights,
		const Optional< ConstTensor > &	biases,
		const char *	name = nullptr
	)

Definition at line 1733 of file Network.cpp.

Referenced by BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), and NetworkImpl::GetGraph().

{
    return AddFullyConnectedLayerImpl(fullyConnectedDescriptor, weights, biases, name);
}

AddFullyConnectedLayer() [2/3]

IConnectableLayer * AddFullyConnectedLayer	(	const FullyConnectedDescriptor &	fullyConnectedDescriptor,
		const ConstTensor &	weights,
		const char *	name = nullptr
	)

Definition at line 1741 of file Network.cpp.

{
    Optional<ConstTensor> biases;
    return AddFullyConnectedLayerImpl(fullyConnectedDescriptor, weights, biases, name);
}

AddFullyConnectedLayer() [3/3]

IConnectableLayer * AddFullyConnectedLayer	(	const FullyConnectedDescriptor &	fullyConnectedDescriptor,
		const ConstTensor &	weights,
		const ConstTensor &	biases,
		const char *	name = nullptr
	)

Definition at line 1749 of file Network.cpp.

{
    Optional<ConstTensor> optionalBiases(biases);
    return AddFullyConnectedLayerImpl(fullyConnectedDescriptor, weights, optionalBiases, name);
}

AddGatherLayer() [1/2]

IConnectableLayer * AddGatherLayer

(

const char *

name = nullptr

)

Definition at line 2251 of file Network.cpp.

Referenced by NetworkImpl::GetGraph().

{
    GatherDescriptor gatherDescriptor{};
    return AddGatherLayer(gatherDescriptor, name);
}

AddGatherLayer() [2/2]

IConnectableLayer * AddGatherLayer	(	const GatherDescriptor &	gatherDescriptor,
		const char *	name = nullptr
	)

Definition at line 2257 of file Network.cpp.

{
    return m_Graph->AddLayer<GatherLayer>(gatherDescriptor, name);
}

AddGreaterLayer()

IConnectableLayer * AddGreaterLayer

(

const char *

name = nullptr

)

Definition at line 2236 of file Network.cpp.

References NetworkImpl::AddComparisonLayer(), and armnn::Greater.

Referenced by NetworkImpl::GetGraph().

{
    return AddComparisonLayer(ComparisonDescriptor(ComparisonOperation::Greater), name);
}

AddInputLayer()

IConnectableLayer * AddInputLayer	(	LayerBindingId	id,
		const char *	name = nullptr
	)

Definition at line 1682 of file Network.cpp.

Referenced by BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), and NetworkImpl::GetGraph().

{
    return m_Graph->AddLayer<InputLayer>(id, name);
}

AddInstanceNormalizationLayer()

IConnectableLayer * AddInstanceNormalizationLayer	(	const InstanceNormalizationDescriptor &	desc,
		const char *	name = nullptr
	)

Definition at line 2009 of file Network.cpp.

Referenced by NetworkImpl::GetGraph().

{
    return m_Graph->AddLayer<InstanceNormalizationLayer>(desc, name);
}

AddL2NormalizationLayer()

IConnectableLayer * AddL2NormalizationLayer	(	const L2NormalizationDescriptor &	desc,
		const char *	name = nullptr
	)

Definition at line 2015 of file Network.cpp.

Referenced by NetworkImpl::GetGraph().

{
    return m_Graph->AddLayer<L2NormalizationLayer>(desc, name);
}

AddLogicalBinaryLayer()

IConnectableLayer * AddLogicalBinaryLayer	(	const LogicalBinaryDescriptor &	logicalBinaryDescriptor,
		const char *	name = nullptr
	)

Definition at line 2499 of file Network.cpp.

Referenced by NetworkImpl::GetGraph().

{
    return m_Graph->AddLayer<LogicalBinaryLayer>(logicalBinaryDescriptor, name);
}

AddLogSoftmaxLayer()

IConnectableLayer * AddLogSoftmaxLayer	(	const LogSoftmaxDescriptor &	logSoftmaxDescriptor,
		const char *	name = nullptr
	)

Definition at line 2021 of file Network.cpp.

Referenced by NetworkImpl::GetGraph().

{
    return m_Graph->AddLayer<LogSoftmaxLayer>(desc, name);
}

AddLstmLayer()

IConnectableLayer * AddLstmLayer	(	const LstmDescriptor &	descriptor,
		const LstmInputParams &	params,
		const char *	name = nullptr
	)

Definition at line 2059 of file Network.cpp.

References LstmLayer::m_BasicParameters, LstmInputParams::m_CellBias, LstmInputParams::m_CellLayerNormWeights, LstmInputParams::m_CellToForgetWeights, LstmInputParams::m_CellToInputWeights, LstmInputParams::m_CellToOutputWeights, LstmInputParams::m_ForgetGateBias, LstmInputParams::m_ForgetLayerNormWeights, LstmInputParams::m_InputGateBias, LstmInputParams::m_InputLayerNormWeights, LstmInputParams::m_InputToCellWeights, LstmInputParams::m_InputToForgetWeights, LstmBasicParameters::m_InputToForgetWeights, LstmInputParams::m_InputToInputWeights, LstmInputParams::m_InputToOutputWeights, LstmInputParams::m_OutputGateBias, LstmInputParams::m_OutputLayerNormWeights, LstmInputParams::m_ProjectionBias, LstmInputParams::m_ProjectionWeights, LstmInputParams::m_RecurrentToCellWeights, LstmInputParams::m_RecurrentToForgetWeights, LstmInputParams::m_RecurrentToInputWeights, and LstmInputParams::m_RecurrentToOutputWeights.

Referenced by armnn::BOOST_AUTO_TEST_CASE(), and NetworkImpl::GetGraph().

{
    const auto layer = m_Graph->AddLayer<LstmLayer>(descriptor, name);

    //Lstm Basic Parameters
    layer->m_BasicParameters.m_InputToForgetWeights =
        std::make_unique<ScopedCpuTensorHandle>(*(params.m_InputToForgetWeights));
    layer->m_BasicParameters.m_InputToCellWeights =
        std::make_unique<ScopedCpuTensorHandle>(*(params.m_InputToCellWeights));
    layer->m_BasicParameters.m_InputToOutputWeights =
        std::make_unique<ScopedCpuTensorHandle>(*(params.m_InputToOutputWeights));
    layer->m_BasicParameters.m_RecurrentToForgetWeights =
        std::make_unique<ScopedCpuTensorHandle>(*(params.m_RecurrentToForgetWeights));
    layer->m_BasicParameters.m_RecurrentToCellWeights =
        std::make_unique<ScopedCpuTensorHandle>(*(params.m_RecurrentToCellWeights));
    layer->m_BasicParameters.m_RecurrentToOutputWeights =
        std::make_unique<ScopedCpuTensorHandle>(*(params.m_RecurrentToOutputWeights));
    layer->m_BasicParameters.m_ForgetGateBias =
            std::make_unique<ScopedCpuTensorHandle>(*(params.m_ForgetGateBias));
    layer->m_BasicParameters.m_CellBias =
            std::make_unique<ScopedCpuTensorHandle>(*(params.m_CellBias));
    layer->m_BasicParameters.m_OutputGateBias =
            std::make_unique<ScopedCpuTensorHandle>(*(params.m_OutputGateBias));

    //Lstm Cifg parameters
    if(!descriptor.m_CifgEnabled)
    {
        if(params.m_InputToInputWeights == nullptr)
        {
            throw InvalidArgumentException("AddLstmLayer: Input To Input Weights cannot be NULL "
                                           "when CIFG is disabled.");
        }
        if(params.m_RecurrentToInputWeights == nullptr)
        {
            throw InvalidArgumentException(
                    "AddLstmLayer: Recurrent To Input Weights cannot be NULL "
                    "when CIFG is disabled.");
        }
        if(params.m_InputGateBias == nullptr)
        {
            throw InvalidArgumentException("AddLstmLayer: Input Gate Bias cannot be NULL "
                                           "when CIFG is disabled.");
        }
        layer->m_CifgParameters.m_InputToInputWeights =
            std::make_unique<ScopedCpuTensorHandle>(*(params.m_InputToInputWeights));
        layer->m_CifgParameters.m_RecurrentToInputWeights =
            std::make_unique<ScopedCpuTensorHandle>(*(params.m_RecurrentToInputWeights));
        layer->m_CifgParameters.m_InputGateBias =
            std::make_unique<ScopedCpuTensorHandle>(*(params.m_InputGateBias));
    }

    //Lstm projection parameters
    if(descriptor.m_ProjectionEnabled)
    {
        if(params.m_ProjectionWeights == nullptr)
        {
            throw InvalidArgumentException("AddLstmLayer: Projection Weights cannot be NULL "
                                           "when projection is enabled.");
        }
        layer->m_ProjectionParameters.m_ProjectionWeights =
            std::make_unique<ScopedCpuTensorHandle>(*(params.m_ProjectionWeights));
        if(params.m_ProjectionBias != nullptr)
        {
            layer->m_ProjectionParameters.m_ProjectionBias =
                std::make_unique<ScopedCpuTensorHandle>(*(params.m_ProjectionBias));
        }
    }

    //Lstm Peephole params
    if(descriptor.m_PeepholeEnabled)
    {
        if(!descriptor.m_CifgEnabled)
        {
            if(params.m_CellToInputWeights == nullptr)
            {
                throw InvalidArgumentException("AddLstmLayer: Cell To Input Weights cannot be NULL "
                                               "when Peephole is enabled and CIFG disabled.");
            }

            layer->m_PeepholeParameters.m_CellToInputWeights =
                std::make_unique<ScopedCpuTensorHandle>(*(params.m_CellToInputWeights));
        }

        if(params.m_CellToForgetWeights == nullptr)
        {
            throw InvalidArgumentException("AddLstmLayer: Cell To Forget Weights cannot be NULL "
                                           "when Peephole is enabled.");
        }
        if(params.m_CellToOutputWeights == nullptr)
        {
            throw InvalidArgumentException("AddLstmLayer: Cell To Output Weights cannot be NULL "
                                           "when Peephole is enabled.");
        }

        layer->m_PeepholeParameters.m_CellToForgetWeights =
            std::make_unique<ScopedCpuTensorHandle>(*(params.m_CellToForgetWeights));
        layer->m_PeepholeParameters.m_CellToOutputWeights =
            std::make_unique<ScopedCpuTensorHandle>(*(params.m_CellToOutputWeights));
    }

    //Lstm Layer Normalization params
    if(descriptor.m_LayerNormEnabled)
    {
        if(!descriptor.m_CifgEnabled)
        {
            if(params.m_InputLayerNormWeights == nullptr)
            {
                throw InvalidArgumentException("AddLstmLayer: Input layer normalization weights cannot be NULL "
                                               "when layer normalization is enabled and CIFG disabled.");
            }
            layer->m_LayerNormParameters.m_InputLayerNormWeights =
                    std::make_unique<ScopedCpuTensorHandle>(*(params.m_InputLayerNormWeights));
        }

        if(params.m_ForgetLayerNormWeights == nullptr)
        {
            throw InvalidArgumentException("AddLstmLayer: Forget layer normalization weights cannot be NULL "
                                           "when layer normalization is enabled.");
        }
        if(params.m_CellLayerNormWeights == nullptr)
        {
            throw InvalidArgumentException("AddLstmLayer: Cell layer normalization weights cannot be NULL "
                                           "when layer normalization is enabled.");
        }
        if(params.m_OutputLayerNormWeights == nullptr)
        {
            throw InvalidArgumentException("AddLstmLayer: Output layer normalization weights cannot be NULL "
                                           "when layer normalization is enabled.");
        }
        layer->m_LayerNormParameters.m_ForgetLayerNormWeights =
                std::make_unique<ScopedCpuTensorHandle>(*(params.m_ForgetLayerNormWeights));
        layer->m_LayerNormParameters.m_CellLayerNormWeights =
                std::make_unique<ScopedCpuTensorHandle>(*(params.m_CellLayerNormWeights));
        layer->m_LayerNormParameters.m_OutputLayerNormWeights =
                std::make_unique<ScopedCpuTensorHandle>(*(params.m_OutputLayerNormWeights));
    }
    return layer;
}

AddMaximumLayer()

IConnectableLayer * AddMaximumLayer

(

const char *

name = nullptr

)

Definition at line 1926 of file Network.cpp.

Referenced by NetworkImpl::GetGraph().

{
    return m_Graph->AddLayer<MaximumLayer>(name);
}

AddMeanLayer()

IConnectableLayer * AddMeanLayer	(	const MeanDescriptor &	meanDescriptor,
		const char *	name = nullptr
	)

Definition at line 2210 of file Network.cpp.

Referenced by NetworkImpl::GetGraph().

{
    return m_Graph->AddLayer<MeanLayer>(meanDescriptor,name);
}

AddMergeLayer()

IConnectableLayer * AddMergeLayer

(

const char *

name = nullptr

)

Definition at line 2263 of file Network.cpp.

Referenced by NetworkImpl::GetGraph().

{
    return m_Graph->AddLayer<MergeLayer>(name);
}

AddMergerLayer()

IConnectableLayer * AddMergerLayer	(	const MergerDescriptor &	mergerDescriptor,
		const char *	name = nullptr
	)

Definition at line 1936 of file Network.cpp.

References NetworkImpl::AddConcatLayer().

Referenced by NetworkImpl::GetGraph().

{
    return AddConcatLayer(mergerDescriptor, name);
}

AddMinimumLayer()

IConnectableLayer * AddMinimumLayer

(

const char *

name = nullptr

)

Definition at line 1931 of file Network.cpp.

Referenced by NetworkImpl::GetGraph().

{
    return m_Graph->AddLayer<MinimumLayer>(name);
}

AddMultiplicationLayer()

IConnectableLayer * AddMultiplicationLayer

(

const char *

name = nullptr

)

Definition at line 1952 of file Network.cpp.

Referenced by BOOST_AUTO_TEST_CASE(), and NetworkImpl::GetGraph().

{
    return m_Graph->AddLayer<MultiplicationLayer>(name);
}

AddNormalizationLayer()

IConnectableLayer * AddNormalizationLayer	(	const NormalizationDescriptor &	normalizationDescriptor,
		const char *	name = nullptr
	)

Definition at line 1902 of file Network.cpp.

Referenced by BOOST_AUTO_TEST_CASE(), and NetworkImpl::GetGraph().

{
    return m_Graph->AddLayer<NormalizationLayer>(normalizationDescriptor, name);
}

AddOutputLayer()

IConnectableLayer * AddOutputLayer	(	LayerBindingId	id,
		const char *	name = nullptr
	)

Definition at line 1957 of file Network.cpp.

Referenced by BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), and NetworkImpl::GetGraph().

{
    return m_Graph->AddLayer<OutputLayer>(id, name);
}

AddPadLayer()

IConnectableLayer * AddPadLayer	(	const PadDescriptor &	padDescriptor,
		const char *	name = nullptr
	)

Definition at line 2215 of file Network.cpp.

Referenced by NetworkImpl::GetGraph().

{
    return m_Graph->AddLayer<PadLayer>(padDescriptor,name);
}

AddPermuteLayer()

IConnectableLayer * AddPermuteLayer	(	const PermuteDescriptor &	permuteDescriptor,
		const char *	name = nullptr
	)

Definition at line 1878 of file Network.cpp.

Referenced by NetworkImpl::GetGraph().

{
    return m_Graph->AddLayer<PermuteLayer>(permuteDescriptor, name);
}

AddPooling2dLayer()

IConnectableLayer * AddPooling2dLayer	(	const Pooling2dDescriptor &	pooling2dDescriptor,
		const char *	name = nullptr
	)

Definition at line 1884 of file Network.cpp.

Referenced by BOOST_AUTO_TEST_CASE(), and NetworkImpl::GetGraph().

{
    return m_Graph->AddLayer<Pooling2dLayer>(pooling2dDescriptor, name);
}

AddPreluLayer()

IConnectableLayer * AddPreluLayer

(

const char *

name = nullptr

)

Definition at line 2273 of file Network.cpp.

Referenced by NetworkImpl::GetGraph().

{
    return m_Graph->AddLayer<PreluLayer>(name);
}

AddQLstmLayer()

IConnectableLayer * AddQLstmLayer	(	const QLstmDescriptor &	descriptor,
		const LstmInputParams &	params,
		const char *	name = nullptr
	)

Definition at line 2357 of file Network.cpp.

References QLstmLayer::m_BasicParameters, LstmInputParams::m_CellBias, LstmInputParams::m_CellLayerNormWeights, LstmInputParams::m_CellToForgetWeights, LstmInputParams::m_CellToInputWeights, LstmInputParams::m_CellToOutputWeights, LstmInputParams::m_ForgetGateBias, LstmInputParams::m_ForgetLayerNormWeights, LstmInputParams::m_InputGateBias, LstmInputParams::m_InputLayerNormWeights, LstmInputParams::m_InputToCellWeights, QLstmBasicParameters::m_InputToForgetWeights, LstmInputParams::m_InputToForgetWeights, LstmInputParams::m_InputToInputWeights, LstmInputParams::m_InputToOutputWeights, LstmInputParams::m_OutputGateBias, LstmInputParams::m_OutputLayerNormWeights, LstmInputParams::m_ProjectionBias, LstmInputParams::m_ProjectionWeights, LstmInputParams::m_RecurrentToCellWeights, LstmInputParams::m_RecurrentToForgetWeights, LstmInputParams::m_RecurrentToInputWeights, and LstmInputParams::m_RecurrentToOutputWeights.

Referenced by armnn::BOOST_AUTO_TEST_CASE(), and NetworkImpl::GetGraph().

{
    const auto layer = m_Graph->AddLayer<QLstmLayer>(descriptor, name);

    // QLstm Basic Parameters
    layer->m_BasicParameters.m_InputToForgetWeights =
            std::make_unique<ScopedCpuTensorHandle>(*(params.m_InputToForgetWeights));
    layer->m_BasicParameters.m_InputToCellWeights =
            std::make_unique<ScopedCpuTensorHandle>(*(params.m_InputToCellWeights));
    layer->m_BasicParameters.m_InputToOutputWeights =
            std::make_unique<ScopedCpuTensorHandle>(*(params.m_InputToOutputWeights));
    layer->m_BasicParameters.m_RecurrentToForgetWeights =
            std::make_unique<ScopedCpuTensorHandle>(*(params.m_RecurrentToForgetWeights));
    layer->m_BasicParameters.m_RecurrentToCellWeights =
            std::make_unique<ScopedCpuTensorHandle>(*(params.m_RecurrentToCellWeights));
    layer->m_BasicParameters.m_RecurrentToOutputWeights =
            std::make_unique<ScopedCpuTensorHandle>(*(params.m_RecurrentToOutputWeights));
    layer->m_BasicParameters.m_ForgetGateBias =
            std::make_unique<ScopedCpuTensorHandle>(*(params.m_ForgetGateBias));
    layer->m_BasicParameters.m_CellBias =
            std::make_unique<ScopedCpuTensorHandle>(*(params.m_CellBias));
    layer->m_BasicParameters.m_OutputGateBias =
            std::make_unique<ScopedCpuTensorHandle>(*(params.m_OutputGateBias));

    // QLstm Cifg parameters
    if(!descriptor.m_CifgEnabled)
    {
        if(params.m_InputToInputWeights == nullptr)
        {
            throw InvalidArgumentException("AddQLstmLayer: Input To Input Weights cannot be NULL");
        }

        if(params.m_RecurrentToInputWeights == nullptr)
        {
            throw InvalidArgumentException(
                    "AddQLstmLayer: Recurrent To Input Weights cannot be NULL");
        }

        if(params.m_InputGateBias == nullptr)
        {
            throw InvalidArgumentException("AddQLstmLayer: Input Gate Bias cannot be NULL");
        }

        layer->m_CifgParameters.m_InputToInputWeights =
                std::make_unique<ScopedCpuTensorHandle>(*(params.m_InputToInputWeights));
        layer->m_CifgParameters.m_RecurrentToInputWeights =
                std::make_unique<ScopedCpuTensorHandle>(*(params.m_RecurrentToInputWeights));
        layer->m_CifgParameters.m_InputGateBias =
                std::make_unique<ScopedCpuTensorHandle>(*(params.m_InputGateBias));
    }

    // QLstm Projection parameters
    if(descriptor.m_ProjectionEnabled)
    {
        if(params.m_ProjectionWeights == nullptr)
        {
            throw InvalidArgumentException("AddQLstmLayer: Projection Weights cannot be NULL");
        }

        layer->m_ProjectionParameters.m_ProjectionWeights =
                std::make_unique<ScopedCpuTensorHandle>(*(params.m_ProjectionWeights));

        // Projection bias is optional even if projection is enabled
        if(params.m_ProjectionWeights != nullptr)
        {
            layer->m_ProjectionParameters.m_ProjectionBias =
                    std::make_unique<ScopedCpuTensorHandle>(*(params.m_ProjectionBias));
        }

    }

    // QLstm Peephole params
    if(descriptor.m_PeepholeEnabled)
    {
        if(params.m_CellToForgetWeights == nullptr)
        {
            throw InvalidArgumentException("AddQLstmLayer: Cell To Forget Weights cannot be NULL");
        }

        if(params.m_CellToOutputWeights == nullptr)
        {
            throw InvalidArgumentException("AddQLstmLayer: Cell To Output Weights cannot be NULL");
        }

        if(!descriptor.m_CifgEnabled)
        {
            if(params.m_CellToInputWeights == nullptr)
            {
                throw InvalidArgumentException("AddQLstmLayer: Cell To Input Weights cannot be NULL");
            }

            layer->m_PeepholeParameters.m_CellToInputWeights =
                    std::make_unique<ScopedCpuTensorHandle>(*(params.m_CellToInputWeights));
        }

        layer->m_PeepholeParameters.m_CellToForgetWeights =
                std::make_unique<ScopedCpuTensorHandle>(*(params.m_CellToForgetWeights));
        layer->m_PeepholeParameters.m_CellToOutputWeights =
                std::make_unique<ScopedCpuTensorHandle>(*(params.m_CellToOutputWeights));
    }

    // QLstm Layer Normalization params
    if(descriptor.m_LayerNormEnabled)
    {
        if(params.m_ForgetLayerNormWeights == nullptr)
        {
            throw InvalidArgumentException("AddQLstmLayer: Forget layer normalization weights cannot be NULL");
        }

        if(params.m_CellLayerNormWeights == nullptr)
        {
            throw InvalidArgumentException("AddQLstmLayer: Cell layer normalization weights cannot be NULL");
        }

        if(params.m_OutputLayerNormWeights == nullptr)
        {
            throw InvalidArgumentException("AddQLstmLayer: Output layer normalization weights cannot be NULL");
        }

        if(!descriptor.m_CifgEnabled)
        {
            if(params.m_InputLayerNormWeights == nullptr)
            {
                throw InvalidArgumentException("AddQLstmLayer: Input layer normalization weights cannot be NULL");
            }

            layer->m_LayerNormParameters.m_InputLayerNormWeights =
                    std::make_unique<ScopedCpuTensorHandle>(*(params.m_InputLayerNormWeights));
        }

        layer->m_LayerNormParameters.m_ForgetLayerNormWeights =
                std::make_unique<ScopedCpuTensorHandle>(*(params.m_ForgetLayerNormWeights));
        layer->m_LayerNormParameters.m_CellLayerNormWeights =
                std::make_unique<ScopedCpuTensorHandle>(*(params.m_CellLayerNormWeights));
        layer->m_LayerNormParameters.m_OutputLayerNormWeights =
                std::make_unique<ScopedCpuTensorHandle>(*(params.m_OutputLayerNormWeights));
    }
    return layer;
}

AddQuantizedLstmLayer()

IConnectableLayer * AddQuantizedLstmLayer	(	const QuantizedLstmInputParams &	params,
		const char *	name = nullptr
	)

Definition at line 2319 of file Network.cpp.

References QuantizedLstmInputParams::GetCellBias(), QuantizedLstmInputParams::GetForgetGateBias(), QuantizedLstmInputParams::GetInputGateBias(), QuantizedLstmInputParams::GetInputToCellWeights(), QuantizedLstmInputParams::GetInputToForgetWeights(), QuantizedLstmInputParams::GetInputToInputWeights(), QuantizedLstmInputParams::GetInputToOutputWeights(), QuantizedLstmInputParams::GetOutputGateBias(), QuantizedLstmInputParams::GetRecurrentToCellWeights(), QuantizedLstmInputParams::GetRecurrentToForgetWeights(), QuantizedLstmInputParams::GetRecurrentToInputWeights(), QuantizedLstmInputParams::GetRecurrentToOutputWeights(), QuantizedLstmParameters::m_InputToInputWeights, and QuantizedLstmLayer::m_QuantizedLstmParameters.

Referenced by armnn::BOOST_AUTO_TEST_CASE(), and NetworkImpl::GetGraph().

{
    const auto layer = m_Graph->AddLayer<QuantizedLstmLayer>(name);

    // InputToX weights
    layer->m_QuantizedLstmParameters.m_InputToInputWeights =
            std::make_unique<ScopedCpuTensorHandle>(params.GetInputToInputWeights());
    layer->m_QuantizedLstmParameters.m_InputToForgetWeights =
            std::make_unique<ScopedCpuTensorHandle>(params.GetInputToForgetWeights());
    layer->m_QuantizedLstmParameters.m_InputToCellWeights =
            std::make_unique<ScopedCpuTensorHandle>(params.GetInputToCellWeights());
    layer->m_QuantizedLstmParameters.m_InputToOutputWeights =
            std::make_unique<ScopedCpuTensorHandle>(params.GetInputToOutputWeights());

    // RecurrentToX weights
    layer->m_QuantizedLstmParameters.m_RecurrentToInputWeights =
            std::make_unique<ScopedCpuTensorHandle>(params.GetRecurrentToInputWeights());
    layer->m_QuantizedLstmParameters.m_RecurrentToForgetWeights =
            std::make_unique<ScopedCpuTensorHandle>(params.GetRecurrentToForgetWeights());
    layer->m_QuantizedLstmParameters.m_RecurrentToCellWeights =
            std::make_unique<ScopedCpuTensorHandle>(params.GetRecurrentToCellWeights());
    layer->m_QuantizedLstmParameters.m_RecurrentToOutputWeights =
            std::make_unique<ScopedCpuTensorHandle>(params.GetRecurrentToOutputWeights());

    // Bias
    layer->m_QuantizedLstmParameters.m_InputGateBias =
            std::make_unique<ScopedCpuTensorHandle>(params.GetInputGateBias());
    layer->m_QuantizedLstmParameters.m_ForgetGateBias =
            std::make_unique<ScopedCpuTensorHandle>(params.GetForgetGateBias());
    layer->m_QuantizedLstmParameters.m_CellBias =
            std::make_unique<ScopedCpuTensorHandle>(params.GetCellBias());
    layer->m_QuantizedLstmParameters.m_OutputGateBias =
            std::make_unique<ScopedCpuTensorHandle>(params.GetOutputGateBias());

    return layer;
}

AddQuantizeLayer()

IConnectableLayer * AddQuantizeLayer

(

const char *

name = nullptr

)

Definition at line 2220 of file Network.cpp.

Referenced by NetworkImpl::GetGraph().

{
    return m_Graph->AddLayer<QuantizeLayer>(name);
}

AddRankLayer()

IConnectableLayer * AddRankLayer

(

const char *

name = nullptr

)

Definition at line 1979 of file Network.cpp.

Referenced by NetworkImpl::GetGraph().

{
    return m_Graph->AddLayer<RankLayer>(name);
}

AddReduceLayer()

IConnectableLayer * AddReduceLayer	(	const ReduceDescriptor &	reduceDescriptor,
		const char *	name = nullptr
	)

Definition at line 1984 of file Network.cpp.

Referenced by NetworkImpl::GetGraph().

{
    return m_Graph->AddLayer<ReduceLayer>(reduceDescriptor, name);
}

AddReshapeLayer()

IConnectableLayer * AddReshapeLayer	(	const ReshapeDescriptor &	reshapeDescriptor,
		const char *	name = nullptr
	)

Definition at line 2036 of file Network.cpp.

Referenced by NetworkImpl::GetGraph().

{
    return m_Graph->AddLayer<ReshapeLayer>(reshapeDescriptor, name);
}

AddResizeBilinearLayer()

IConnectableLayer * AddResizeBilinearLayer	(	const ResizeBilinearDescriptor &	resizeDesc,
		const char *	name = nullptr
	)

Definition at line 1990 of file Network.cpp.

References armnn::Bilinear, ResizeBilinearDescriptor::m_AlignCorners, ResizeDescriptor::m_AlignCorners, ResizeBilinearDescriptor::m_DataLayout, ResizeDescriptor::m_DataLayout, ResizeBilinearDescriptor::m_HalfPixelCenters, ResizeDescriptor::m_HalfPixelCenters, ResizeDescriptor::m_Method, ResizeBilinearDescriptor::m_TargetHeight, ResizeDescriptor::m_TargetHeight, ResizeBilinearDescriptor::m_TargetWidth, and ResizeDescriptor::m_TargetWidth.

Referenced by NetworkImpl::GetGraph().

{
    ResizeDescriptor resizeDescriptor;
    resizeDescriptor.m_Method           = ResizeMethod::Bilinear;
    resizeDescriptor.m_DataLayout       = descriptor.m_DataLayout;
    resizeDescriptor.m_TargetWidth      = descriptor.m_TargetWidth;
    resizeDescriptor.m_TargetHeight     = descriptor.m_TargetHeight;
    resizeDescriptor.m_AlignCorners     = descriptor.m_AlignCorners;
    resizeDescriptor.m_HalfPixelCenters = descriptor.m_HalfPixelCenters;

    return m_Graph->AddLayer<ResizeLayer>(resizeDescriptor, name);
}

AddResizeLayer()

IConnectableLayer * AddResizeLayer	(	const ResizeDescriptor &	resizeDescriptor,
		const char *	name = nullptr
	)

Definition at line 2004 of file Network.cpp.

Referenced by NetworkImpl::GetGraph().

{
    return m_Graph->AddLayer<ResizeLayer>(resizeDescriptor, name);
}

AddRsqrtLayer()

IConnectableLayer * AddRsqrtLayer

(

const char *

name = nullptr

)

Definition at line 2246 of file Network.cpp.

References NetworkImpl::AddElementwiseUnaryLayer(), and armnn::Rsqrt.

Referenced by NetworkImpl::GetGraph().

{
    return AddElementwiseUnaryLayer(ElementwiseUnaryDescriptor(UnaryOperation::Rsqrt), name);
}

AddSliceLayer()

IConnectableLayer * AddSliceLayer	(	const SliceDescriptor &	sliceDescriptor,
		const char *	name = nullptr
	)

Definition at line 1909 of file Network.cpp.

Referenced by NetworkImpl::GetGraph().

{
    return m_Graph->AddLayer<SliceLayer>(sliceDescriptor, name);
}

AddSoftmaxLayer()

IConnectableLayer * AddSoftmaxLayer	(	const SoftmaxDescriptor &	softmaxDescriptor,
		const char *	name = nullptr
	)

Definition at line 1914 of file Network.cpp.

Referenced by BOOST_AUTO_TEST_CASE(), and NetworkImpl::GetGraph().

{
    return m_Graph->AddLayer<SoftmaxLayer>(softmaxDescriptor, name);
}

AddSpaceToBatchNdLayer()

IConnectableLayer * AddSpaceToBatchNdLayer	(	const SpaceToBatchNdDescriptor &	spaceToBatchNdDescriptor,
		const char *	name = nullptr
	)

Definition at line 2042 of file Network.cpp.

Referenced by NetworkImpl::GetGraph().

{
    return m_Graph->AddLayer<SpaceToBatchNdLayer>(spaceToBatchNdDescriptor, name);
}

AddSpaceToDepthLayer()

IConnectableLayer * AddSpaceToDepthLayer	(	const SpaceToDepthDescriptor &	spaceToDepthDescriptor,
		const char *	name = nullptr
	)

Definition at line 2048 of file Network.cpp.

Referenced by NetworkImpl::GetGraph().

{
    return m_Graph->AddLayer<SpaceToDepthLayer>(spaceToDepthDescriptor, name);
}

AddSplitterLayer()

IConnectableLayer * AddSplitterLayer	(	const ViewsDescriptor &	splitterDescriptor,
		const char *	name = nullptr
	)

Definition at line 1920 of file Network.cpp.

Referenced by BOOST_AUTO_TEST_CASE(), and NetworkImpl::GetGraph().

{
    return m_Graph->AddLayer<SplitterLayer>(splitterDescriptor, name);
}

AddStackLayer()

IConnectableLayer * AddStackLayer	(	const StackDescriptor &	stackDescriptor,
		const char *	name = nullptr
	)

Definition at line 2306 of file Network.cpp.

Referenced by NetworkImpl::GetGraph().

{
    return m_Graph->AddLayer<StackLayer>(stackDescriptor, name);
}

AddStandInLayer()

IConnectableLayer * AddStandInLayer	(	const StandInDescriptor &	descriptor,
		const char *	name = nullptr
	)

Definition at line 2313 of file Network.cpp.

Referenced by BOOST_AUTO_TEST_CASE(), and NetworkImpl::GetGraph().

{
    return m_Graph->AddLayer<StandInLayer>(desc, name);
}

AddStridedSliceLayer()

IConnectableLayer * AddStridedSliceLayer	(	const StridedSliceDescriptor &	stridedSliceDescriptor,
		const char *	name = nullptr
	)

Definition at line 2230 of file Network.cpp.

Referenced by NetworkImpl::GetGraph().

{
    return m_Graph->AddLayer<StridedSliceLayer>(stridedSliceDescriptor, name);
}

AddSubtractionLayer()

IConnectableLayer * AddSubtractionLayer

(

const char *

name = nullptr

)

Definition at line 2205 of file Network.cpp.

Referenced by NetworkImpl::GetGraph().

{
    return m_Graph->AddLayer<SubtractionLayer>(name);
}

AddSwitchLayer()

IConnectableLayer * AddSwitchLayer

(

const char *

name = nullptr

)

Definition at line 2268 of file Network.cpp.

Referenced by NetworkImpl::GetGraph().

{
    return m_Graph->AddLayer<SwitchLayer>(name);
}

AddTransposeConvolution2dLayer()

IConnectableLayer * AddTransposeConvolution2dLayer	(	const TransposeConvolution2dDescriptor &	descriptor,
		const ConstTensor &	weights,
		const Optional< ConstTensor > &	biases,
		const char *	name = nullptr
	)

Definition at line 2278 of file Network.cpp.

References OptionalBase::has_value(), TransposeConvolution2dDescriptor::m_BiasEnabled, TransposeConvolution2dLayer::m_Weight, and OptionalReferenceSwitch< std::is_reference< T >::value, T >::value().

Referenced by NetworkImpl::GetGraph().

{
    if (descriptor.m_BiasEnabled && !biases.has_value())
    {
        throw InvalidArgumentException("AddTransposeConvolution2dLayer: Biases cannot be empty");
    }

    const auto layer = m_Graph->AddLayer<TransposeConvolution2dLayer>(descriptor, name);

    layer->m_Weight = std::make_unique<ScopedCpuTensorHandle>(weights);

    if (descriptor.m_BiasEnabled)
    {
        layer->m_Bias = std::make_unique<ScopedCpuTensorHandle>(biases.value());
    }

    return layer;
}

AddTransposeLayer()

IConnectableLayer * AddTransposeLayer	(	const TransposeDescriptor &	transposeDescriptor,
		const char *	name = nullptr
	)

Definition at line 2300 of file Network.cpp.

Referenced by NetworkImpl::GetGraph().

{
    return m_Graph->AddLayer<TransposeLayer>(transposeDescriptor, name);
}

ExecuteStrategy()

void ExecuteStrategy

(

IStrategy &

strategy

)

const

Definition at line 2513 of file Network.cpp.

References NetworkImpl::GetGraph().

Referenced by NetworkImpl::GetGraph().

{
    for (auto layer : GetGraph())
    {
        layer->ExecuteStrategy(strategy);
    };
}

GetGraph()

const Graph& GetGraph ( ) const

inline

Definition at line 37 of file Network.hpp.

References NetworkImpl::Accept(), NetworkImpl::AddAbsLayer(), NetworkImpl::AddActivationLayer(), NetworkImpl::AddAdditionLayer(), NetworkImpl::AddArgMinMaxLayer(), NetworkImpl::AddBatchNormalizationLayer(), NetworkImpl::AddBatchToSpaceNdLayer(), NetworkImpl::AddComparisonLayer(), NetworkImpl::AddConcatLayer(), NetworkImpl::AddConstantLayer(), NetworkImpl::AddConvolution2dLayer(), NetworkImpl::AddDepthToSpaceLayer(), NetworkImpl::AddDepthwiseConvolution2dLayer(), NetworkImpl::AddDequantizeLayer(), NetworkImpl::AddDetectionPostProcessLayer(), NetworkImpl::AddDivisionLayer(), NetworkImpl::AddElementwiseUnaryLayer(), NetworkImpl::AddEqualLayer(), NetworkImpl::AddFillLayer(), NetworkImpl::AddFloorLayer(), NetworkImpl::AddFullyConnectedLayer(), NetworkImpl::AddGatherLayer(), NetworkImpl::AddGreaterLayer(), NetworkImpl::AddInputLayer(), NetworkImpl::AddInstanceNormalizationLayer(), NetworkImpl::AddL2NormalizationLayer(), NetworkImpl::AddLogicalBinaryLayer(), NetworkImpl::AddLogSoftmaxLayer(), NetworkImpl::AddLstmLayer(), NetworkImpl::AddMaximumLayer(), NetworkImpl::AddMeanLayer(), NetworkImpl::AddMergeLayer(), NetworkImpl::AddMergerLayer(), NetworkImpl::AddMinimumLayer(), NetworkImpl::AddMultiplicationLayer(), NetworkImpl::AddNormalizationLayer(), NetworkImpl::AddOutputLayer(), NetworkImpl::AddPadLayer(), NetworkImpl::AddPermuteLayer(), NetworkImpl::AddPooling2dLayer(), NetworkImpl::AddPreluLayer(), NetworkImpl::AddQLstmLayer(), NetworkImpl::AddQuantizedLstmLayer(), NetworkImpl::AddQuantizeLayer(), NetworkImpl::AddRankLayer(), NetworkImpl::AddReduceLayer(), NetworkImpl::AddReshapeLayer(), NetworkImpl::AddResizeBilinearLayer(), NetworkImpl::AddResizeLayer(), NetworkImpl::AddRsqrtLayer(), NetworkImpl::AddSliceLayer(), NetworkImpl::AddSoftmaxLayer(), NetworkImpl::AddSpaceToBatchNdLayer(), NetworkImpl::AddSpaceToDepthLayer(), NetworkImpl::AddSplitterLayer(), NetworkImpl::AddStackLayer(), NetworkImpl::AddStandInLayer(), NetworkImpl::AddStridedSliceLayer(), NetworkImpl::AddSubtractionLayer(), NetworkImpl::AddSwitchLayer(), NetworkImpl::AddTransposeConvolution2dLayer(), NetworkImpl::AddTransposeLayer(), anchors(), ARMNN_DEPRECATED_MSG, NetworkImpl::ExecuteStrategy(), and NetworkImpl::PrintGraph().

Referenced by NetworkImpl::Accept(), BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), NetworkImpl::ExecuteStrategy(), and armnn::GetInputTensorInfo().

{ return *m_Graph; }

PrintGraph()

Status PrintGraph

(

)

Definition at line 1676 of file Network.cpp.

References armnn::Success.

Referenced by BOOST_AUTO_TEST_CASE(), and NetworkImpl::GetGraph().

{
    m_Graph->Print();
    return Status::Success;
}

---

The documentation for this class was generated from the following files:

• src/armnn/Network.hpp
• src/armnn/Network.cpp