1 //
2 //  CoreMLArgMax.hpp
3 //  MNN
4 //
5 //  Created by MNN on 2021/03/31.
6 //  Copyright © 2018, Alibaba Group Holding Limited
7 //
8 
9 #ifndef MNN_COREMLARGMAX_HPP
10 #define MNN_COREMLARGMAX_HPP
11 
12 #include "CoreMLCommonExecution.hpp"
13 #include "CoreMLBackend.hpp"
14 
15 namespace MNN {
16 
17 class CoreMLArgMax : public CoreMLCommonExecution {
18 public:
19     CoreMLArgMax(Backend *b, const Op *op, const std::vector<Tensor *> &inputs, const std::vector<Tensor *> &outputs);
20     ErrorCode onResize(const std::vector<Tensor *> &inputs, const std::vector<Tensor *> &outputs);
21     virtual ~CoreMLArgMax() = default;
22 };
23 } // namespace MNN
24 
25 #endif // MNN_COREMLARGMAX_HPP
26