13 #include <doctest/doctest.h> 27 Connect(inputLayer, dequantizeLayer, inputInfo, 0, 0);
28 Connect(dequantizeLayer, outputLayer, outputInfo, 0, 0);
33 template<armnn::DataType ArmnnType,
typename T = armnn::ResolveType<ArmnnType>>
34 void DequantizeEndToEndLayerTestImpl(
const std::vector<BackendId>& backends,
36 const std::vector<T>& input,
37 const std::vector<float>& expectedOutput,
52 std::map<int, std::vector<T>> inputTensorData = { { 0, input } };
53 std::map<int, std::vector<float>> expectedOutputData = { { 0, expectedOutput } };
55 EndToEndLayerTestImpl<ArmnnType, armnn::DataType::Float32>(
56 move(net), inputTensorData, expectedOutputData, backends);
59 template<armnn::DataType ArmnnType,
typename T = armnn::ResolveType<ArmnnType>>
60 void DequantizeEndToEndSimple(
const std::vector<BackendId>& backends)
63 std::vector<T> inputData = std::vector<T>(
71 std::vector<float> expectedOutputData = std::vector<float>(
73 1.0f, 2.0f, 3.0f, 4.0f,
74 5.0f, 6.0f, 7.0f, 8.0f,
75 9.0f, 10.0f, 11.0f, 12.0f,
76 13.0f, 14.0f, 15.0f, 16.0f
78 DequantizeEndToEndLayerTestImpl<ArmnnType>(backends, tensorShape, inputData, expectedOutputData, 0.5f, 0);
81 template<armnn::DataType ArmnnType,
typename T = armnn::ResolveType<ArmnnType>>
82 void DequantizeEndToEndOffset(
const std::vector<BackendId>& backends)
85 std::vector<T> inputData = std::vector<T>(
93 std::vector<float> expectedOutputData = std::vector<float>(
95 1.0f, 2.0f, 3.0f, 4.0f,
96 5.0f, 6.0f, 7.0f, 8.0f,
97 9.0f, 10.0f, 11.0f, 12.0f,
98 13.0f, 14.0f, 15.0f, 16.0f
100 DequantizeEndToEndLayerTestImpl<ArmnnType>(backends, tensorShape, inputData, expectedOutputData, 0.5f, 1);
Interface for a layer that is connectable to other layers via InputSlots and OutputSlots.
void SetQuantizationScale(float scale)
void SetQuantizationOffset(int32_t offset)
void Connect(armnn::IConnectableLayer *from, armnn::IConnectableLayer *to, const armnn::TensorInfo &tensorInfo, unsigned int fromIndex, unsigned int toIndex)
std::unique_ptr< INetwork, void(*)(INetwork *network)> INetworkPtr
static INetworkPtr Create(NetworkOptions networkOptions={})