e25f20f7a3
commit 3f4938648950a7f3bf9a19c320ca9fae7c52de20 Author: sophgo-forum-service <forum_service@sophgo.com> Date: Mon May 13 13:44:23 2024 +0800 [feat] cviruntime opensource for cv18xx soc. - a4b6a3, add cumsum and gatherelements_pt.
23 lines
582 B
C++
23 lines
582 B
C++
/*
|
|
* Copyright (C) Cvitek Co., Ltd. 2019-2020. All rights reserved.
|
|
*/
|
|
#ifndef ROI_ALIGN_OP_H_
|
|
#define ROI_ALIGN_OP_H_
|
|
|
|
#include "tpuc/CustomOp.h"
|
|
|
|
namespace cvi {
|
|
|
|
class ROIAlignOp : public CustomOp {
|
|
public:
|
|
ROIAlignOp(OpParam ¶m) : CustomOp(param) {}
|
|
|
|
void interpretFp32(std::vector<std::shared_ptr<std::vector<float>>> &operand_tensors,
|
|
std::vector<std::vector<int64_t>> &operand_shapes,
|
|
std::shared_ptr<std::vector<float>> &result_tensor,
|
|
std::vector<int64_t> &result_shape);
|
|
};
|
|
|
|
} // namespace cvi
|
|
#endif
|