StridedSliceAsyncEndToEndTest.hpp File Reference

#include <ResolveType.hpp>
#include <armnn/IWorkingMemHandle.hpp>
#include <armnn/INetwork.hpp>
#include <armnn/IAsyncExecutionCallback.hpp>
#include <AsyncExecutionCallback.hpp>
#include <backendsCommon/test/CommonTestUtils.hpp>
#include <boost/test/unit_test.hpp>
#include <vector>

Go to the source code of this file.

Namespaces
	armnn
	Copyright (c) 2021 ARM Limited and Contributors.
	armnn::experimental

Functions
template<DataType ArmnnIType, DataType ArmnnOType, typename TInput = ResolveType <ArmnnIType>, typename TOutput = ResolveType <ArmnnOType>>
void	AsyncThreadedEndToEndTestImpl (INetworkPtr network, const std::vector< std::map< int, std::vector< TInput >>> &inputTensorData, const std::vector< std::map< int, std::vector< TOutput >>> &expectedOutputData, std::vector< BackendId > backends, const size_t numberOfInferences, float tolerance=0.000001f)
template<DataType ArmnnIType, DataType ArmnnOType, typename TInput = ResolveType<ArmnnIType>, typename TOutput = ResolveType<ArmnnOType>>
void	AsyncEndToEndTestImpl (INetworkPtr network, const std::map< int, std::vector< TInput >> &inputTensorData, const std::map< int, std::vector< TOutput >> &expectedOutputData, std::vector< BackendId > backends, float tolerance=0.000001f, size_t numThreads=0)
template<typename armnn::DataType DataType>
INetworkPtr	CreateStridedSliceNetwork (const TensorShape &inputShape, const TensorShape &outputShape, const std::vector< int > &beginData, const std::vector< int > &endData, const std::vector< int > &stridesData, int beginMask=0, int endMask=0, int shrinkAxisMask=0, int ellipsisMask=0, int newAxisMask=0, const float qScale=1.0f, const int32_t qOffset=0)
template<armnn::DataType ArmnnType>
void	StridedSlicedEndToEndTest (const std::vector< BackendId > &backends)
template<armnn::DataType ArmnnType>
void	AsyncScheduledStridedSlicedEndToEndTest (const std::vector< BackendId > &backends)
template<armnn::DataType ArmnnType>
void	AsyncScheduledStridedSlicedMultiThreadedEndToEndTest (const std::vector< BackendId > &backends)
template<armnn::DataType ArmnnType>
void	StridedSlicedMultiThreadedEndToEndTest (const std::vector< BackendId > &backends)