20.02
|
#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 <boost/algorithm/string.hpp>
#include <boost/test/unit_test.hpp>
Go to the source code of this file.
void ArgMinMaxInferOutputShape1dTest | ( | ) |
Definition at line 87 of file InferOutputTests.hpp.
References ArgMinMaxInferOutputShapeImpl(), BOOST_CHECK(), and ArgMinMaxDescriptor::m_Axis.
void ArgMinMaxInferOutputShape2dTest | ( | ) |
Definition at line 68 of file InferOutputTests.hpp.
References ArgMinMaxInferOutputShapeImpl(), BOOST_CHECK(), and ArgMinMaxDescriptor::m_Axis.
void ArgMinMaxInferOutputShape3dTest | ( | ) |
Definition at line 49 of file InferOutputTests.hpp.
References ArgMinMaxInferOutputShapeImpl(), BOOST_CHECK(), and ArgMinMaxDescriptor::m_Axis.
void ArgMinMaxInferOutputShape4dTest | ( | ) |
Definition at line 30 of file InferOutputTests.hpp.
References ArgMinMaxInferOutputShapeImpl(), BOOST_CHECK(), and ArgMinMaxDescriptor::m_Axis.
void ArgMinMaxInferOutputShapeImpl | ( | const armnn::ArgMinMaxDescriptor | descriptor, |
const std::vector< armnn::TensorShape > & | inputShapes, | ||
std::vector< armnn::TensorShape > & | outputShapes | ||
) |
Definition at line 21 of file InferOutputTests.hpp.
References Graph::AddLayer(), and ArgMinMaxLayer::InferOutputShapes().
Referenced by ArgMinMaxInferOutputShape1dTest(), ArgMinMaxInferOutputShape2dTest(), ArgMinMaxInferOutputShape3dTest(), and ArgMinMaxInferOutputShape4dTest().
void BatchToSpaceInferOutputShapeTest | ( | ) |
Definition at line 106 of file InferOutputTests.hpp.
References Graph::AddLayer(), BOOST_CHECK(), BatchToSpaceNdLayer::InferOutputShapes(), BatchToSpaceNdDescriptor::m_BlockShape, BatchToSpaceNdDescriptor::m_Crops, BatchToSpaceNdDescriptor::m_DataLayout, and armnn::NHWC.
void Convolution2dInferOutputShapeTest | ( | ) |
Definition at line 436 of file InferOutputTests.hpp.
References Graph::AddLayer(), BOOST_CHECK(), 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.
void CreatePreluLayerHelper | ( | armnn::Graph & | graph, |
const armnn::TensorShape & | inputShape, | ||
const armnn::TensorShape & | alphaShape, | ||
const armnn::TensorShape & | outputShape | ||
) |
Definition at line 239 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 353 of file InferOutputTests.hpp.
References Graph::AddLayer(), Connect(), and armnn::Float32.
Referenced by StackValidateTensorShapesFromInputsMatchTest(), and StackValidateTensorShapesFromInputsNoMatchTest().
void DepthwiseConvolution2dInferOutputShapeTest | ( | ) |
Definition at line 500 of file InferOutputTests.hpp.
References Graph::AddLayer(), BOOST_CHECK(), 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.
void PreluInferOutputShapeAlphaBiggerTest | ( | ) |
Definition at line 199 of file InferOutputTests.hpp.
References BOOST_CHECK(), and PreluInferOutputShapeImpl().
void PreluInferOutputShapeImpl | ( | const std::vector< armnn::TensorShape > & | inputShapes, |
std::vector< armnn::TensorShape > & | outputShapes | ||
) |
Definition at line 151 of file InferOutputTests.hpp.
References Graph::AddLayer(), and PreluLayer::InferOutputShapes().
Referenced by PreluInferOutputShapeAlphaBiggerTest(), PreluInferOutputShapeInputBiggerTest(), PreluInferOutputShapeNoMatchTest(), and PreluInferOutputShapeSameDimsTest().
void PreluInferOutputShapeInputBiggerTest | ( | ) |
Definition at line 179 of file InferOutputTests.hpp.
References BOOST_CHECK(), and PreluInferOutputShapeImpl().
void PreluInferOutputShapeNoMatchTest | ( | ) |
Definition at line 219 of file InferOutputTests.hpp.
References BOOST_CHECK(), and PreluInferOutputShapeImpl().
void PreluInferOutputShapeSameDimsTest | ( | ) |
Definition at line 159 of file InferOutputTests.hpp.
References BOOST_CHECK(), and PreluInferOutputShapeImpl().
void PreluValidateTensorShapesFromInputsMatchTest | ( | ) |
Definition at line 261 of file InferOutputTests.hpp.
References CreatePreluLayerHelper(), and Graph::InferTensorInfos().
void PreluValidateTensorShapesFromInputsNoMatchTest | ( | ) |
Definition at line 272 of file InferOutputTests.hpp.
References CreatePreluLayerHelper(), and Graph::InferTensorInfos().
void QuantizedLstmInferOutputShapeImpl | ( | const std::vector< armnn::TensorShape > & | inputShapes, |
std::vector< armnn::TensorShape > & | outputShapes | ||
) |
Definition at line 534 of file InferOutputTests.hpp.
References Graph::AddLayer(), and QuantizedLstmLayer::InferOutputShapes().
Referenced by QuantizedLstmInferOutputShapeTest().
void QuantizedLstmInferOutputShapeTest | ( | ) |
Definition at line 542 of file InferOutputTests.hpp.
References BOOST_CHECK(), and QuantizedLstmInferOutputShapeImpl().
void SpaceToDepthInferOutputShapeTest | ( | ) |
Definition at line 129 of file InferOutputTests.hpp.
References Graph::AddLayer(), BOOST_CHECK(), SpaceToDepthLayer::InferOutputShapes(), SpaceToDepthDescriptor::m_BlockSize, SpaceToDepthDescriptor::m_DataLayout, and armnn::NHWC.
void StackInferOutputShapeFromInputsMatchTest | ( | ) |
Definition at line 292 of file InferOutputTests.hpp.
References BOOST_CHECK(), StackDescriptor::m_Axis, StackDescriptor::m_InputShape, StackDescriptor::m_NumInputs, and StackInferOutputShapeImpl().
void StackInferOutputShapeFromInputsNoMatchTest | ( | ) |
Definition at line 322 of file InferOutputTests.hpp.
References BOOST_CHECK(), StackDescriptor::m_Axis, StackDescriptor::m_InputShape, StackDescriptor::m_NumInputs, and StackInferOutputShapeImpl().
void StackInferOutputShapeImpl | ( | const armnn::StackDescriptor | descriptor, |
const std::vector< armnn::TensorShape > & | inputShapes, | ||
std::vector< armnn::TensorShape > & | outputShapes | ||
) |
Definition at line 283 of file InferOutputTests.hpp.
References Graph::AddLayer(), and StackLayer::InferOutputShapes().
Referenced by StackInferOutputShapeFromInputsMatchTest(), and StackInferOutputShapeFromInputsNoMatchTest().
void StackValidateTensorShapesFromInputsMatchTest | ( | ) |
Definition at line 384 of file InferOutputTests.hpp.
References CreateStackLayerHelper(), Graph::InferTensorInfos(), StackDescriptor::m_Axis, StackDescriptor::m_InputShape, and StackDescriptor::m_NumInputs.
void StackValidateTensorShapesFromInputsNoMatchTest | ( | ) |
Definition at line 410 of file InferOutputTests.hpp.
References CreateStackLayerHelper(), Graph::InferTensorInfos(), StackDescriptor::m_Axis, StackDescriptor::m_InputShape, and StackDescriptor::m_NumInputs.
void TransposeConvolution2dInferOutputShapeTest | ( | ) |
Definition at line 469 of file InferOutputTests.hpp.
References Graph::AddLayer(), BOOST_CHECK(), 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.