22.08
|
#include <TestUtils.hpp>
#include <Graph.hpp>
#include <layers/ArgMinMaxLayer.hpp>
#include <layers/BatchToSpaceNdLayer.hpp>
#include <layers/SpaceToDepthLayer.hpp>
#include <layers/PreluLayer.hpp>
#include <layers/StackLayer.hpp>
#include <doctest/doctest.h>
Go to the source code of this file.
void ArgMinMaxInferOutputShape1dTest | ( | ) |
Definition at line 85 of file InferOutputTests.hpp.
References ArgMinMaxInferOutputShapeImpl(), and ArgMinMaxDescriptor::m_Axis.
Referenced by TEST_SUITE().
void ArgMinMaxInferOutputShape2dTest | ( | ) |
Definition at line 66 of file InferOutputTests.hpp.
References ArgMinMaxInferOutputShapeImpl(), and ArgMinMaxDescriptor::m_Axis.
Referenced by TEST_SUITE().
void ArgMinMaxInferOutputShape3dTest | ( | ) |
Definition at line 47 of file InferOutputTests.hpp.
References ArgMinMaxInferOutputShapeImpl(), and ArgMinMaxDescriptor::m_Axis.
Referenced by TEST_SUITE().
void ArgMinMaxInferOutputShape4dTest | ( | ) |
Definition at line 28 of file InferOutputTests.hpp.
References ArgMinMaxInferOutputShapeImpl(), and ArgMinMaxDescriptor::m_Axis.
Referenced by TEST_SUITE().
void ArgMinMaxInferOutputShapeImpl | ( | const armnn::ArgMinMaxDescriptor | descriptor, |
const std::vector< armnn::TensorShape > & | inputShapes, | ||
std::vector< armnn::TensorShape > & | outputShapes | ||
) |
Definition at line 19 of file InferOutputTests.hpp.
References Graph::AddLayer(), and ArgMinMaxLayer::InferOutputShapes().
Referenced by ArgMinMaxInferOutputShape1dTest(), ArgMinMaxInferOutputShape2dTest(), ArgMinMaxInferOutputShape3dTest(), and ArgMinMaxInferOutputShape4dTest().
void BatchToSpaceInferOutputShapeTest | ( | ) |
Definition at line 104 of file InferOutputTests.hpp.
References Graph::AddLayer(), BatchToSpaceNdLayer::InferOutputShapes(), BatchToSpaceNdDescriptor::m_BlockShape, BatchToSpaceNdDescriptor::m_Crops, BatchToSpaceNdDescriptor::m_DataLayout, and armnn::NHWC.
Referenced by TEST_SUITE().
void Convolution2dInferOutputShapeTest | ( | ) |
Definition at line 434 of file InferOutputTests.hpp.
References Graph::AddLayer(), Convolution2dLayer::InferOutputShapes(), Convolution2dDescriptor::m_DataLayout, Convolution2dDescriptor::m_DilationX, Convolution2dDescriptor::m_DilationY, Convolution2dDescriptor::m_PadBottom, Convolution2dDescriptor::m_PadLeft, Convolution2dDescriptor::m_PadRight, Convolution2dDescriptor::m_PadTop, Convolution2dDescriptor::m_StrideX, Convolution2dDescriptor::m_StrideY, and armnn::NCHW.
Referenced by TEST_SUITE().
void Convolution3dInferOutputShapeTest | ( | ) |
Definition at line 467 of file InferOutputTests.hpp.
References Graph::AddLayer(), Convolution3dLayer::InferOutputShapes(), Convolution3dDescriptor::m_DataLayout, Convolution3dDescriptor::m_DilationX, Convolution3dDescriptor::m_DilationY, Convolution3dDescriptor::m_DilationZ, Convolution3dDescriptor::m_PadBack, Convolution3dDescriptor::m_PadBottom, Convolution3dDescriptor::m_PadFront, Convolution3dDescriptor::m_PadLeft, Convolution3dDescriptor::m_PadRight, Convolution3dDescriptor::m_PadTop, Convolution3dDescriptor::m_StrideX, Convolution3dDescriptor::m_StrideY, Convolution3dDescriptor::m_StrideZ, and armnn::NDHWC.
Referenced by TEST_SUITE().
void CreatePreluLayerHelper | ( | armnn::Graph & | graph, |
const armnn::TensorShape & | inputShape, | ||
const armnn::TensorShape & | alphaShape, | ||
const armnn::TensorShape & | outputShape | ||
) |
Definition at line 237 of file InferOutputTests.hpp.
References Graph::AddLayer(), Connect(), and armnn::Float32.
Referenced by PreluValidateTensorShapesFromInputsMatchTest(), and PreluValidateTensorShapesFromInputsNoMatchTest().
void CreateStackLayerHelper | ( | armnn::Graph & | graph, |
const armnn::StackDescriptor & | descriptor, | ||
const std::vector< armnn::TensorShape > & | inputShapes, | ||
const armnn::TensorShape & | outputShape | ||
) |
Definition at line 351 of file InferOutputTests.hpp.
References Graph::AddLayer(), Connect(), and armnn::Float32.
Referenced by StackValidateTensorShapesFromInputsMatchTest(), and StackValidateTensorShapesFromInputsNoMatchTest().
void DepthwiseConvolution2dInferOutputShapeTest | ( | ) |
Definition at line 535 of file InferOutputTests.hpp.
References Graph::AddLayer(), DepthwiseConvolution2dLayer::InferOutputShapes(), DepthwiseConvolution2dDescriptor::m_DataLayout, DepthwiseConvolution2dDescriptor::m_DilationX, DepthwiseConvolution2dDescriptor::m_DilationY, DepthwiseConvolution2dDescriptor::m_PadBottom, DepthwiseConvolution2dDescriptor::m_PadLeft, DepthwiseConvolution2dDescriptor::m_PadRight, DepthwiseConvolution2dDescriptor::m_PadTop, DepthwiseConvolution2dDescriptor::m_StrideX, DepthwiseConvolution2dDescriptor::m_StrideY, and armnn::NCHW.
Referenced by TEST_SUITE().
void Pooling3dInferOutputShapeTest | ( | ) |
Definition at line 568 of file InferOutputTests.hpp.
References Graph::AddLayer(), Pooling3dLayer::InferOutputShapes(), Pooling3dDescriptor::m_DataLayout, Pooling3dDescriptor::m_PadBack, Pooling3dDescriptor::m_PadBottom, Pooling3dDescriptor::m_PadFront, Pooling3dDescriptor::m_PadLeft, Pooling3dDescriptor::m_PadRight, Pooling3dDescriptor::m_PadTop, Pooling3dDescriptor::m_PoolDepth, Pooling3dDescriptor::m_PoolHeight, Pooling3dDescriptor::m_PoolType, Pooling3dDescriptor::m_PoolWidth, Pooling3dDescriptor::m_StrideX, Pooling3dDescriptor::m_StrideY, Pooling3dDescriptor::m_StrideZ, armnn::Max, and armnn::NDHWC.
Referenced by TEST_SUITE().
void PreluInferOutputShapeAlphaBiggerTest | ( | ) |
Definition at line 197 of file InferOutputTests.hpp.
References PreluInferOutputShapeImpl().
Referenced by TEST_SUITE().
void PreluInferOutputShapeImpl | ( | const std::vector< armnn::TensorShape > & | inputShapes, |
std::vector< armnn::TensorShape > & | outputShapes | ||
) |
Definition at line 149 of file InferOutputTests.hpp.
References Graph::AddLayer(), and PreluLayer::InferOutputShapes().
Referenced by PreluInferOutputShapeAlphaBiggerTest(), PreluInferOutputShapeInputBiggerTest(), PreluInferOutputShapeNoMatchTest(), and PreluInferOutputShapeSameDimsTest().
void PreluInferOutputShapeInputBiggerTest | ( | ) |
Definition at line 177 of file InferOutputTests.hpp.
References PreluInferOutputShapeImpl().
Referenced by TEST_SUITE().
void PreluInferOutputShapeNoMatchTest | ( | ) |
Definition at line 217 of file InferOutputTests.hpp.
References PreluInferOutputShapeImpl().
Referenced by TEST_SUITE().
void PreluInferOutputShapeSameDimsTest | ( | ) |
Definition at line 157 of file InferOutputTests.hpp.
References PreluInferOutputShapeImpl().
Referenced by TEST_SUITE().
void PreluValidateTensorShapesFromInputsMatchTest | ( | ) |
Definition at line 259 of file InferOutputTests.hpp.
References CreatePreluLayerHelper(), and Graph::InferTensorInfos().
Referenced by TEST_SUITE().
void PreluValidateTensorShapesFromInputsNoMatchTest | ( | ) |
Definition at line 270 of file InferOutputTests.hpp.
References CreatePreluLayerHelper(), and Graph::InferTensorInfos().
Referenced by TEST_SUITE().
void QLstmInferOutputShapeImpl | ( | const armnn::QLstmDescriptor | descriptor, |
const std::vector< armnn::TensorShape > & | inputShapes, | ||
std::vector< armnn::TensorShape > & | outputShapes | ||
) |
Definition at line 603 of file InferOutputTests.hpp.
References Graph::AddLayer(), and QLstmLayer::InferOutputShapes().
Referenced by QLstmInferOutputShapeTest().
void QLstmInferOutputShapeTest | ( | ) |
Definition at line 612 of file InferOutputTests.hpp.
References QLstmDescriptor::m_CifgEnabled, QLstmDescriptor::m_PeepholeEnabled, QLstmDescriptor::m_ProjectionEnabled, and QLstmInferOutputShapeImpl().
Referenced by TEST_SUITE().
void QuantizedLstmInferOutputShapeImpl | ( | const std::vector< armnn::TensorShape > & | inputShapes, |
std::vector< armnn::TensorShape > & | outputShapes | ||
) |
Definition at line 660 of file InferOutputTests.hpp.
References Graph::AddLayer(), and QuantizedLstmLayer::InferOutputShapes().
Referenced by QuantizedLstmInferOutputShapeTest().
void QuantizedLstmInferOutputShapeTest | ( | ) |
Definition at line 668 of file InferOutputTests.hpp.
References QuantizedLstmInferOutputShapeImpl().
Referenced by TEST_SUITE().
void SpaceToDepthInferOutputShapeTest | ( | ) |
Definition at line 127 of file InferOutputTests.hpp.
References Graph::AddLayer(), SpaceToDepthLayer::InferOutputShapes(), SpaceToDepthDescriptor::m_BlockSize, SpaceToDepthDescriptor::m_DataLayout, and armnn::NHWC.
Referenced by TEST_SUITE().
void StackInferOutputShapeFromInputsMatchTest | ( | ) |
Definition at line 290 of file InferOutputTests.hpp.
References StackDescriptor::m_Axis, StackDescriptor::m_InputShape, StackDescriptor::m_NumInputs, and StackInferOutputShapeImpl().
Referenced by TEST_SUITE().
void StackInferOutputShapeFromInputsNoMatchTest | ( | ) |
Definition at line 320 of file InferOutputTests.hpp.
References StackDescriptor::m_Axis, StackDescriptor::m_InputShape, StackDescriptor::m_NumInputs, and StackInferOutputShapeImpl().
Referenced by TEST_SUITE().
void StackInferOutputShapeImpl | ( | const armnn::StackDescriptor | descriptor, |
const std::vector< armnn::TensorShape > & | inputShapes, | ||
std::vector< armnn::TensorShape > & | outputShapes | ||
) |
Definition at line 281 of file InferOutputTests.hpp.
References Graph::AddLayer(), and StackLayer::InferOutputShapes().
Referenced by StackInferOutputShapeFromInputsMatchTest(), and StackInferOutputShapeFromInputsNoMatchTest().
void StackValidateTensorShapesFromInputsMatchTest | ( | ) |
Definition at line 382 of file InferOutputTests.hpp.
References CreateStackLayerHelper(), Graph::InferTensorInfos(), StackDescriptor::m_Axis, StackDescriptor::m_InputShape, and StackDescriptor::m_NumInputs.
Referenced by TEST_SUITE().
void StackValidateTensorShapesFromInputsNoMatchTest | ( | ) |
Definition at line 408 of file InferOutputTests.hpp.
References CreateStackLayerHelper(), Graph::InferTensorInfos(), StackDescriptor::m_Axis, StackDescriptor::m_InputShape, and StackDescriptor::m_NumInputs.
Referenced by TEST_SUITE().
void TransposeConvolution2dInferOutputShapeTest | ( | ) |
Definition at line 504 of file InferOutputTests.hpp.
References Graph::AddLayer(), TransposeConvolution2dLayer::InferOutputShapes(), TransposeConvolution2dDescriptor::m_DataLayout, TransposeConvolution2dDescriptor::m_PadBottom, TransposeConvolution2dDescriptor::m_PadLeft, TransposeConvolution2dDescriptor::m_PadRight, TransposeConvolution2dDescriptor::m_PadTop, TransposeConvolution2dDescriptor::m_StrideX, TransposeConvolution2dDescriptor::m_StrideY, and armnn::NCHW.
Referenced by TEST_SUITE().