21.02
|
#include "../TestUtils.hpp"
#include <BFloat16.hpp>
#include <Optimizer.hpp>
#include <boost/test/unit_test.hpp>
Go to the source code of this file.
Functions | |
BOOST_AUTO_TEST_CASE (ConvertConstantsFloatToBFloatTest) | |
BOOST_AUTO_TEST_CASE (ConvertConstantsBFloatToFloatTest) | |
BOOST_AUTO_TEST_CASE | ( | ConvertConstantsFloatToBFloatTest | ) |
Definition at line 18 of file ConvertConstantsBFloatTests.cpp.
References Graph::AddLayer(), armnn::BFloat16, OutputSlot::Connect(), armnn::Float32, Layer::GetOutputSlot(), armnn::info, FullyConnectedLayer::m_Weight, armnn::MakeOptimizations(), Optimizer::Pass(), and OutputSlot::SetTensorInfo().
BOOST_AUTO_TEST_CASE | ( | ConvertConstantsBFloatToFloatTest | ) |
Definition at line 71 of file ConvertConstantsBFloatTests.cpp.
References Graph::AddLayer(), armnn::BFloat16, BOOST_AUTO_TEST_SUITE_END(), OutputSlot::Connect(), FloatingPointConverter::ConvertFloat32ToBFloat16(), armnn::Float32, Layer::GetOutputSlot(), armnn::info, FullyConnectedLayer::m_Weight, armnn::MakeOptimizations(), Optimizer::Pass(), and OutputSlot::SetTensorInfo().