25 Connect(inputLayer, dequantizeLayer, inputInfo, 0, 0);
26 Connect(dequantizeLayer, outputLayer, outputInfo, 0, 0);
31 template<armnn::DataType ArmnnType,
typename T = armnn::ResolveType<ArmnnType>>
32 void DequantizeEndToEndLayerTestImpl(
const std::vector<BackendId>& backends,
34 const std::vector<T>& input,
35 const std::vector<float>& expectedOutput,
48 BOOST_TEST_CHECKPOINT(
"create a network");
50 std::map<int, std::vector<T>> inputTensorData = { { 0, input } };
51 std::map<int, std::vector<float>> expectedOutputData = { { 0, expectedOutput } };
53 EndToEndLayerTestImpl<ArmnnType, armnn::DataType::Float32>(
54 move(net), inputTensorData, expectedOutputData, backends);
57 template<armnn::DataType ArmnnType,
typename T = armnn::ResolveType<ArmnnType>>
58 void DequantizeEndToEndSimple(
const std::vector<BackendId>& backends)
61 std::vector<T> inputData = std::vector<T>(
69 std::vector<float> expectedOutputData = std::vector<float>(
71 1.0f, 2.0f, 3.0f, 4.0f,
72 5.0f, 6.0f, 7.0f, 8.0f,
73 9.0f, 10.0f, 11.0f, 12.0f,
74 13.0f, 14.0f, 15.0f, 16.0f
76 DequantizeEndToEndLayerTestImpl<ArmnnType>(backends, tensorShape, inputData, expectedOutputData, 0.5f, 0);
79 template<armnn::DataType ArmnnType,
typename T = armnn::ResolveType<ArmnnType>>
80 void DequantizeEndToEndOffset(
const std::vector<BackendId>& backends)
83 std::vector<T> inputData = std::vector<T>(
91 std::vector<float> expectedOutputData = std::vector<float>(
93 1.0f, 2.0f, 3.0f, 4.0f,
94 5.0f, 6.0f, 7.0f, 8.0f,
95 9.0f, 10.0f, 11.0f, 12.0f,
96 13.0f, 14.0f, 15.0f, 16.0f
98 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()