20.08
|
Go to the source code of this file.
Functions | |
BOOST_AUTO_TEST_CASE (Fp32NetworkToBf16OptimizationNoConversionTest) | |
BOOST_AUTO_TEST_CASE (Fp32NetworkToBf16OptimizationConv2DTest) | |
BOOST_AUTO_TEST_CASE (Fp32NetworkToBf16OptimizationFullyConnectedTest) | |
BOOST_AUTO_TEST_CASE | ( | Fp32NetworkToBf16OptimizationNoConversionTest | ) |
Definition at line 15 of file Fp32NetworkToBf16ConverterTests.cpp.
References Graph::AddLayer(), Graph::cbegin(), Graph::cend(), CheckSequence(), OutputSlot::Connect(), armnn::Float32, Layer::GetOutputSlot(), armnn::MakeOptimizations(), Optimizer::Pass(), and OutputSlot::SetTensorInfo().
BOOST_AUTO_TEST_CASE | ( | Fp32NetworkToBf16OptimizationConv2DTest | ) |
Definition at line 45 of file Fp32NetworkToBf16ConverterTests.cpp.
References Graph::AddLayer(), armnn::BFloat16, Graph::cbegin(), Graph::cend(), CheckSequence(), OutputSlot::Connect(), armnn::Float32, TensorInfo::GetDataType(), Layer::GetOutputSlot(), Convolution2dLayer::m_Weight, armnn::MakeOptimizations(), Optimizer::Pass(), and OutputSlot::SetTensorInfo().
BOOST_AUTO_TEST_CASE | ( | Fp32NetworkToBf16OptimizationFullyConnectedTest | ) |
Definition at line 115 of file Fp32NetworkToBf16ConverterTests.cpp.
References Graph::AddLayer(), armnn::BFloat16, BOOST_AUTO_TEST_SUITE_END(), Graph::cbegin(), Graph::cend(), CheckSequence(), OutputSlot::Connect(), armnn::Float32, TensorInfo::GetDataType(), Layer::GetOutputSlot(), FullyConnectedLayer::m_Weight, armnn::MakeOptimizations(), Optimizer::Pass(), and OutputSlot::SetTensorInfo().