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,
53 std::map<int, std::vector<T>> inputTensorData = { { 0, input } };
54 std::map<int, std::vector<float>> expectedOutputData = { { 0, expectedOutput } };
56 EndToEndLayerTestImpl<ArmnnType, armnn::DataType::Float32>(
57 move(net), inputTensorData, expectedOutputData, backends);
60 template<armnn::DataType ArmnnType,
typename T = armnn::ResolveType<ArmnnType>>
61 void DequantizeEndToEndSimple(
const std::vector<BackendId>& backends)
64 std::vector<T> inputData = std::vector<T>(
72 std::vector<float> expectedOutputData = std::vector<float>(
74 1.0f, 2.0f, 3.0f, 4.0f,
75 5.0f, 6.0f, 7.0f, 8.0f,
76 9.0f, 10.0f, 11.0f, 12.0f,
77 13.0f, 14.0f, 15.0f, 16.0f
79 DequantizeEndToEndLayerTestImpl<ArmnnType>(backends, tensorShape, inputData, expectedOutputData, 0.5f, 0);
82 template<armnn::DataType ArmnnType,
typename T = armnn::ResolveType<ArmnnType>>
83 void DequantizeEndToEndOffset(
const std::vector<BackendId>& backends)
86 std::vector<T> inputData = std::vector<T>(
94 std::vector<float> expectedOutputData = std::vector<float>(
96 1.0f, 2.0f, 3.0f, 4.0f,
97 5.0f, 6.0f, 7.0f, 8.0f,
98 9.0f, 10.0f, 11.0f, 12.0f,
99 13.0f, 14.0f, 15.0f, 16.0f
101 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 SetConstant(const bool IsConstant=true)
Marks the data corresponding to this tensor info as constant.
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={})