Function CreateReduceTfLiteModel

delegate/test/ReduceTestHelper.hpp:18–139

{

std::vector<char> CreateReduceTfLiteModel(tflite::BuiltinOperator reduceOperatorCode,
                                          tflite::TensorType tensorType,
                                          std::vector<int32_t>& input0TensorShape,
                                          std::vector<int32_t>& input1TensorShape,
                                          const std::vector <int32_t>& outputTensorShape,
                                          std::vector<int32_t>& axisData,
                                          const bool keepDims,
                                          float quantScale = 1.0f,
                                          int quantOffset  = 0,
                                          bool kTfLiteNoQuantizationForQuantized = false)
{
    using namespace tflite;
    flatbuffers::FlatBufferBuilder flatBufferBuilder;

    flatbuffers::Offset<tflite::Buffer> buffers[4] = {
            CreateBuffer(flatBufferBuilder),
            CreateBuffer(flatBufferBuilder),
            CreateBuffer(flatBufferBuilder,
                         flatBufferBuilder.CreateVector(reinterpret_cast<const uint8_t*>(axisData.data()),
                                                        sizeof(int32_t) * axisData.size())),
            CreateBuffer(flatBufferBuilder)
    };

    flatbuffers::Offset<tflite::QuantizationParameters> quantizationParametersAxis
            = CreateQuantizationParameters(flatBufferBuilder);

    flatbuffers::Offset<tflite::QuantizationParameters> quantizationParameters;

    if (kTfLiteNoQuantizationForQuantized && reduceOperatorCode == BuiltinOperator_REDUCE_PROD)
    {
        if ((quantScale == 1 || quantScale == 0) && quantOffset == 0)
        {
            // Creates quantization parameter with quantization.type = kTfLiteNoQuantization
            quantizationParameters = CreateQuantizationParameters(flatBufferBuilder);
        }
        else
        {
            // Creates quantization parameter with quantization.type != kTfLiteNoQuantization
            quantizationParameters = CreateQuantizationParameters(
                    flatBufferBuilder,
                    0,
                    0,
                    flatBufferBuilder.CreateVector<float>({quantScale}),
                    flatBufferBuilder.CreateVector<int64_t>({quantOffset}));
        }
    }
    else
    {
        quantizationParameters = CreateQuantizationParameters(
                flatBufferBuilder,
                0,
                0,
                flatBufferBuilder.CreateVector<float>({quantScale}),
                flatBufferBuilder.CreateVector<int64_t>({quantOffset}));
    }

    std::array<flatbuffers::Offset<Tensor>, 3> tensors;
    tensors[0] = CreateTensor(flatBufferBuilder,