Method ParseStridedSlice

src/armnnTfLiteParser/TfLiteParser.cpp:2625–2710

}

void TfLiteParserImpl::ParseStridedSlice(size_t subgraphIndex, size_t operatorIndex)
{
    CHECK_MODEL(m_Model, subgraphIndex, operatorIndex);

    auto inputs = GetInputs(m_Model, subgraphIndex, operatorIndex);
    CHECK_VALID_SIZE(inputs.size(), 4);

    auto outputs = GetOutputs(m_Model, subgraphIndex, operatorIndex);
    CHECK_VALID_SIZE(outputs.size(), 1);

    const auto& operatorPtr = m_Model->subgraphs[subgraphIndex]->operators[operatorIndex];
    const auto* options = operatorPtr->builtin_options.AsStridedSliceOptions();

    StridedSliceDescriptor desc;
    desc.m_BeginMask = options->begin_mask;
    desc.m_EllipsisMask = options->ellipsis_mask;
    desc.m_EndMask = options->end_mask;
    desc.m_NewAxisMask = options->new_axis_mask;
    desc.m_ShrinkAxisMask = options->shrink_axis_mask;
    desc.m_DataLayout = armnn::DataLayout::NHWC;

    armnn::TensorInfo beginTensorInfo = InputTensorInfo(subgraphIndex, operatorIndex, 1);
    BufferRawPtr beginBufferPtr = GetBuffer(m_Model, inputs[1]->buffer);
    ValidateBuffer(beginBufferPtr, beginTensorInfo, "begin");

    std::vector<int> begin(beginTensorInfo.GetNumElements());
    if (beginBufferPtr->data.data() != nullptr)
    {
        ::memcpy(begin.data(), beginBufferPtr->data.data(), beginTensorInfo.GetNumBytes());
    }
    else
    {
        throw ParseException("ParseStridedSlice: Invalid input - the begin vector is null");
    }

    armnn::TensorInfo endTensorInfo = InputTensorInfo(subgraphIndex, operatorIndex, 2);
    BufferRawPtr endBufferPtr = GetBuffer(m_Model, inputs[2]->buffer);
    ValidateBuffer(endBufferPtr, endTensorInfo, "end");

    std::vector<int> end(endTensorInfo.GetNumElements());
    if (endBufferPtr->data.data() != nullptr)
    {
        ::memcpy(end.data(), endBufferPtr->data.data(), endTensorInfo.GetNumBytes());
    }
    else
    {
        throw ParseException("ParseStridedSlice: Invalid input - the end vector is null");
    }

    armnn::TensorInfo strideTensorInfo = InputTensorInfo(subgraphIndex, operatorIndex, 3);
    BufferRawPtr strideBufferPtr = GetBuffer(m_Model, inputs[3]->buffer);
    ValidateBuffer(strideBufferPtr, strideTensorInfo, "stride");

    std::vector<int> stride(strideTensorInfo.GetNumElements());

    if (strideBufferPtr->data.data() != nullptr)
    {
        ::memcpy(stride.data(), strideBufferPtr->data.data(), strideTensorInfo.GetNumBytes());