diff options
Diffstat (limited to 'delegate/src/test/QuantizationTestHelper.hpp')
-rw-r--r-- | delegate/src/test/QuantizationTestHelper.hpp | 11 |
1 files changed, 7 insertions, 4 deletions
diff --git a/delegate/src/test/QuantizationTestHelper.hpp b/delegate/src/test/QuantizationTestHelper.hpp index e4155040cd..a8b102271a 100644 --- a/delegate/src/test/QuantizationTestHelper.hpp +++ b/delegate/src/test/QuantizationTestHelper.hpp @@ -1,5 +1,5 @@ // -// Copyright © 2020 Arm Ltd and Contributors. All rights reserved. +// Copyright © 2020, 2023 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // @@ -31,7 +31,10 @@ std::vector<char> CreateQuantizationTfLiteModel(tflite::BuiltinOperator quantiza flatbuffers::FlatBufferBuilder flatBufferBuilder; std::vector<flatbuffers::Offset<tflite::Buffer>> buffers; - buffers.push_back(CreateBuffer(flatBufferBuilder, flatBufferBuilder.CreateVector({}))); + buffers.push_back(CreateBuffer(flatBufferBuilder)); + buffers.push_back(CreateBuffer(flatBufferBuilder)); + buffers.push_back(CreateBuffer(flatBufferBuilder)); + auto quantizationParameters = CreateQuantizationParameters(flatBufferBuilder, @@ -46,14 +49,14 @@ std::vector<char> CreateQuantizationTfLiteModel(tflite::BuiltinOperator quantiza flatBufferBuilder.CreateVector<int32_t>(inputTensorShape.data(), inputTensorShape.size()), inputTensorType, - 0, + 1, flatBufferBuilder.CreateString("input"), quantizationParameters); tensors[1] = CreateTensor(flatBufferBuilder, flatBufferBuilder.CreateVector<int32_t>(outputTensorShape.data(), outputTensorShape.size()), outputTensorType, - 0, + 2, flatBufferBuilder.CreateString("output"), quantizationParameters); |