#include "backend/cpu/CPUBackend.hpp" namespace MNN { class CPUGatherND : public Execution { public: CPUGatherND(Backend* bn) : Execution(bn) { //Do nothing; } virtual ErrorCode onResize(const std::vector &inputs, const std::vector &outputs) override; virtual ErrorCode onExecute(const std::vector &inputs, const std::vector &outputs) override; private: std::vector mDimsToCount; int mSliceN = 0; int mSliceSize = 0; }; }