21.11
|
Classes | |
class | AsyncCallbackManager |
class | AsyncExecutionCallback |
class | IAsyncExecutionCallback |
class | IWorkingMemHandle |
class | Threadpool |
struct | WorkingMemDescriptor |
class | WorkingMemHandle |
Typedefs | |
using | IAsyncExecutionCallbackPtr = std::shared_ptr< IAsyncExecutionCallback > |
using | InferenceId = uint64_t |
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=1) |
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, size_t numThreads) |
template<armnn::DataType ArmnnType> | |
void | StridedSlicedMultiThreadedEndToEndTest (const std::vector< BackendId > &backends) |
using IAsyncExecutionCallbackPtr = std::shared_ptr<IAsyncExecutionCallback> |
Definition at line 17 of file IAsyncExecutionCallback.hpp.
using InferenceId = uint64_t |
Definition at line 24 of file AsyncExecutionCallback.hpp.
void armnn::experimental::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 = 1 |
||
) |
Definition at line 124 of file StridedSliceAsyncEndToEndTest.hpp.
References IRuntime::Create(), AsyncCallbackManager::GetNotifiedCallback(), armnn::Optimize(), TensorInfo::SetConstant(), armnn::Success, and armnn::Undefined.
void armnn::experimental::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 |
||
) |
Definition at line 30 of file StridedSliceAsyncEndToEndTest.hpp.
References IRuntime::Create(), armnn::Optimize(), TensorInfo::SetConstant(), and armnn::Undefined.
INetworkPtr armnn::experimental::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 |
||
) |
Definition at line 223 of file StridedSliceAsyncEndToEndTest.hpp.
References Connect(), INetwork::Create(), StridedSliceDescriptor::m_Begin, StridedSliceDescriptor::m_BeginMask, StridedSliceDescriptor::m_EllipsisMask, StridedSliceDescriptor::m_End, StridedSliceDescriptor::m_EndMask, StridedSliceDescriptor::m_NewAxisMask, StridedSliceDescriptor::m_ShrinkAxisMask, and StridedSliceDescriptor::m_Stride.
void armnn::experimental::StridedSlicedEndToEndTest | ( | const std::vector< BackendId > & | backends, |
size_t | numThreads | ||
) |
Definition at line 264 of file StridedSliceAsyncEndToEndTest.hpp.
void armnn::experimental::StridedSlicedMultiThreadedEndToEndTest | ( | const std::vector< BackendId > & | backends | ) |
Definition at line 318 of file StridedSliceAsyncEndToEndTest.hpp.