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.
13 lines
343 B
C++
13 lines
343 B
C++
/*
|
|
* Copyright (C) Cvitek Co., Ltd. 2019-2020. All rights reserved.
|
|
*/
|
|
#include <runtime/cpu_function.hpp>
|
|
#include "SoftmaxOpRuntime.hpp"
|
|
#include "ROIAlignOpRuntime.hpp"
|
|
|
|
REGISTER_OP_RUNTIME_FUNCS(
|
|
{(char *)"mysoftmax", SoftmaxOpRuntime::open},
|
|
{(char *)"roialign", ROIAlignOpRuntime::open}
|
|
// add more custom op runtime func here.
|
|
);
|