Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
MegEngine 天元
MegEngine
提交
972a8d54
MegEngine
项目概览
MegEngine 天元
/
MegEngine
1 年多 前同步成功
通知
404
Star
4705
Fork
582
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
DevOps
流水线
流水线任务
计划
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
MegEngine
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
DevOps
DevOps
流水线
流水线任务
计划
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
流水线任务
提交
Issue看板
提交
972a8d54
编写于
5月 20, 2022
作者:
M
Megvii Engine Team
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
feat(ci): add model compatibility check in ci
GitOrigin-RevId: d17c21ab4fd43005ecfc5a27c7488c6ef866445e
上级
e6943017
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
30 addition
and
0 deletion
+30
-0
src/opr/impl/dnn/dnn.sereg.h
src/opr/impl/dnn/dnn.sereg.h
+30
-0
未找到文件。
src/opr/impl/dnn/dnn.sereg.h
浏览文件 @
972a8d54
...
@@ -284,6 +284,34 @@ struct OprMaker<opr::LSQBackward, 5> {
...
@@ -284,6 +284,34 @@ struct OprMaker<opr::LSQBackward, 5> {
}
}
};
};
template
<
>
struct
OprMaker
<
opr
::
RNNCellForward
,
6
>
{
using
Param
=
opr
::
RNNCellForward
::
Param
;
static
cg
::
OperatorNodeBase
*
make
(
const
Param
&
param
,
const
cg
::
VarNodeArray
&
i
,
ComputingGraph
&
graph
,
const
OperatorNodeConfig
&
config
)
{
MGB_MARK_USED_VAR
(
graph
);
return
opr
::
RNNCellForward
::
make
(
i
[
0
],
i
[
1
],
i
[
2
],
i
[
3
],
i
[
4
],
i
[
5
],
param
,
config
)
.
node
()
->
owner_opr
();
}
};
template
<
>
struct
OprMaker
<
opr
::
LSTMCellForward
,
7
>
{
using
Param
=
opr
::
LSTMCellForward
::
Param
;
static
cg
::
OperatorNodeBase
*
make
(
const
Param
&
param
,
const
cg
::
VarNodeArray
&
i
,
ComputingGraph
&
graph
,
const
OperatorNodeConfig
&
config
)
{
MGB_MARK_USED_VAR
(
graph
);
return
opr
::
LSTMCellForward
::
make
(
i
[
0
],
i
[
1
],
i
[
2
],
i
[
3
],
i
[
4
],
i
[
5
],
i
[
6
],
param
,
config
)
.
node
()
->
owner_opr
();
}
};
template
<
>
template
<
>
struct
OprMaker
<
opr
::
RNNBackward
,
7
>
{
struct
OprMaker
<
opr
::
RNNBackward
,
7
>
{
using
Param
=
opr
::
RNNBackward
::
Param
;
using
Param
=
opr
::
RNNBackward
::
Param
;
...
@@ -718,6 +746,8 @@ MGB_SEREG_OPR(LSQ, 4);
...
@@ -718,6 +746,8 @@ MGB_SEREG_OPR(LSQ, 4);
MGB_SEREG_OPR
(
LSQBackward
,
5
);
MGB_SEREG_OPR
(
LSQBackward
,
5
);
MGB_SEREG_OPR
(
LayerNorm
,
0
);
MGB_SEREG_OPR
(
LayerNorm
,
0
);
MGB_SEREG_OPR
(
LayerNormBackward
,
0
);
MGB_SEREG_OPR
(
LayerNormBackward
,
0
);
MGB_SEREG_OPR
(
RNNCellForward
,
6
);
MGB_SEREG_OPR
(
LSTMCellForward
,
7
);
MGB_SEREG_OPR
(
RNNForward
,
3
);
MGB_SEREG_OPR
(
RNNForward
,
3
);
MGB_SEREG_OPR
(
RNNBackward
,
7
);
MGB_SEREG_OPR
(
RNNBackward
,
7
);
MGB_SEREG_OPR
(
LSTMForward
,
4
);
MGB_SEREG_OPR
(
LSTMForward
,
4
);
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录