//
//  QNNStridedSlice.hpp
//  MNN
//
//  Created by MNN on b'2025/04/10'.
//  Copyright © 2018, Alibaba Group Holding Limited
//

#ifndef MNN_QNNSTRIDEDSLICE_HPP
#define MNN_QNNSTRIDEDSLICE_HPP

#include "QNNCommonExecution.hpp"

namespace MNN {
namespace QNN {
#ifdef ENABLE_QNN_ONLINE_FINALIZE

class QNNStridedSlice : public QNNCommonExecution {
public:
    QNNStridedSlice(Backend *backend, const Op *op);
    virtual ErrorCode onEncode(const std::vector<Tensor *> &inputs, const std::vector<Tensor *> &outputs) override;
private:
    void computeRangesType0(const std::vector<Tensor *> &inputs, std::vector<int> & beginRaw, std::vector<int> & endRaw, std::vector<int> & strideRaw);
    void computeRangesType1(const std::vector<Tensor *> &inputs, std::vector<int> & beginRaw, std::vector<int> & endRaw, std::vector<int> & strideRaw);
    uint32_t computeMask(uint32_t rawMask, int dim, Tensor::DimensionType dimType);
    int mInputDim;
    Tensor::DimensionType mDimType;
    bool mIsSlice = false;
};
#endif
} // end namespace QNN
} // end namespace MNN

#endif // end MNN_QNNSTRIDEDSLICE_HPP