1 // 2 // CPUTopKV2.hpp 3 // MNN 4 // 5 // Created by MNN on 2018/08/28. 6 // Copyright © 2018, Alibaba Group Holding Limited 7 // 8 9 #ifndef CPUTOPKV2_HPP 10 #define CPUTOPKV2_HPP 11 12 #include "core/Execution.hpp" 13 #include "MNN_generated.h" 14 15 namespace MNN { 16 class CPUTopKV2 : public Execution { 17 public: 18 CPUTopKV2(Backend *b); 19 virtual ~CPUTopKV2() = default; 20 virtual ErrorCode onExecute(const std::vector<Tensor *> &inputs, const std::vector<Tensor *> &outputs) override; 21 22 private: 23 }; 24 } // namespace MNN 25 26 #endif // CPUTOPKV2_HPP 27