Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
慢慢CG
Mace
提交
5efffb7c
Mace
项目概览
慢慢CG
/
Mace
与 Fork 源项目一致
Fork自
Xiaomi / Mace
通知
1
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
DevOps
流水线
流水线任务
计划
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
Mace
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
DevOps
DevOps
流水线
流水线任务
计划
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
流水线任务
提交
Issue看板
提交
5efffb7c
编写于
3月 22, 2018
作者:
L
Liangliang He
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
stash
上级
25df7e53
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
21 addition
and
18 deletion
+21
-18
mace/core/operator.cc
mace/core/operator.cc
+19
-16
mace/python/tools/str2vec_maps.cc.tmpl
mace/python/tools/str2vec_maps.cc.tmpl
+2
-2
未找到文件。
mace/core/operator.cc
浏览文件 @
5efffb7c
...
@@ -64,6 +64,7 @@ std::unique_ptr<OperatorBase> OperatorRegistry::CreateOperator(
...
@@ -64,6 +64,7 @@ std::unique_ptr<OperatorBase> OperatorRegistry::CreateOperator(
namespace
ops
{
namespace
ops
{
// Keep in lexicographical order
extern
void
Register_Activation
(
OperatorRegistry
*
op_registry
);
extern
void
Register_Activation
(
OperatorRegistry
*
op_registry
);
extern
void
Register_AddN
(
OperatorRegistry
*
op_registry
);
extern
void
Register_AddN
(
OperatorRegistry
*
op_registry
);
extern
void
Register_BatchNorm
(
OperatorRegistry
*
op_registry
);
extern
void
Register_BatchNorm
(
OperatorRegistry
*
op_registry
);
...
@@ -74,27 +75,29 @@ extern void Register_ChannelShuffle(OperatorRegistry *op_registry);
...
@@ -74,27 +75,29 @@ extern void Register_ChannelShuffle(OperatorRegistry *op_registry);
extern
void
Register_Concat
(
OperatorRegistry
*
op_registry
);
extern
void
Register_Concat
(
OperatorRegistry
*
op_registry
);
extern
void
Register_Conv2D
(
OperatorRegistry
*
op_registry
);
extern
void
Register_Conv2D
(
OperatorRegistry
*
op_registry
);
extern
void
Register_DepthwiseConv2d
(
OperatorRegistry
*
op_registry
);
extern
void
Register_DepthwiseConv2d
(
OperatorRegistry
*
op_registry
);
extern
void
Register_Eltwise
(
OperatorRegistry
*
op_registry
);
extern
void
Register_FoldedBatchNorm
(
OperatorRegistry
*
op_registry
);
extern
void
Register_FoldedBatchNorm
(
OperatorRegistry
*
op_registry
);
extern
void
Register_FullyConnected
(
OperatorRegistry
*
op_registry
);
extern
void
Register_FusedConv2D
(
OperatorRegistry
*
op_registry
);
extern
void
Register_FusedConv2D
(
OperatorRegistry
*
op_registry
);
extern
void
Register_GlobalAvgPooling
(
OperatorRegistry
*
op_registry
);
extern
void
Register_GlobalAvgPooling
(
OperatorRegistry
*
op_registry
);
extern
void
Register_ImageToBuffer
(
OperatorRegistry
*
op_registry
);
extern
void
Register_ImageToBuffer
(
OperatorRegistry
*
op_registry
);
extern
void
Register_MatMul
(
OperatorRegistry
*
op_registry
);
extern
void
Register_Pooling
(
OperatorRegistry
*
op_registry
);
extern
void
Register_Pooling
(
OperatorRegistry
*
op_registry
);
extern
void
Register_Proposal
(
OperatorRegistry
*
op_registry
);
extern
void
Register_PSROIAlign
(
OperatorRegistry
*
op_registry
);
extern
void
Register_Reshape
(
OperatorRegistry
*
op_registry
);
extern
void
Register_ResizeBilinear
(
OperatorRegistry
*
op_registry
);
extern
void
Register_ResizeBilinear
(
OperatorRegistry
*
op_registry
);
extern
void
Register_Slice
(
OperatorRegistry
*
op_registry
);
extern
void
Register_Softmax
(
OperatorRegistry
*
op_registry
);
extern
void
Register_Softmax
(
OperatorRegistry
*
op_registry
);
extern
void
Register_SpaceToBatchND
(
OperatorRegistry
*
op_registry
);
extern
void
Register_SpaceToBatchND
(
OperatorRegistry
*
op_registry
);
extern
void
Register_MatMul
(
OperatorRegistry
*
op_registry
);
extern
void
Register_WinogradTransform
(
OperatorRegistry
*
op_registry
);
extern
void
Register_WinogradInverseTransform
(
OperatorRegistry
*
op_registry
);
extern
void
Register_WinogradInverseTransform
(
OperatorRegistry
*
op_registry
);
extern
void
Register_Reshape
(
OperatorRegistry
*
op_registry
);
extern
void
Register_WinogradTransform
(
OperatorRegistry
*
op_registry
);
extern
void
Register_Eltwise
(
OperatorRegistry
*
op_registry
);
extern
void
Register_FullyConnected
(
OperatorRegistry
*
op_registry
);
extern
void
Register_Slice
(
OperatorRegistry
*
op_registry
);
extern
void
Register_Proposal
(
OperatorRegistry
*
op_registry
);
extern
void
Register_PSROIAlign
(
OperatorRegistry
*
op_registry
);
}
// namespace ops
}
// namespace ops
OperatorRegistry
::
OperatorRegistry
()
{
OperatorRegistry
::
OperatorRegistry
()
{
// Keep in lexicographical order
ops
::
Register_Activation
(
this
);
ops
::
Register_Activation
(
this
);
ops
::
Register_AddN
(
this
);
ops
::
Register_AddN
(
this
);
ops
::
Register_BatchNorm
(
this
);
ops
::
Register_BatchNorm
(
this
);
...
@@ -105,23 +108,23 @@ OperatorRegistry::OperatorRegistry() {
...
@@ -105,23 +108,23 @@ OperatorRegistry::OperatorRegistry() {
ops
::
Register_Concat
(
this
);
ops
::
Register_Concat
(
this
);
ops
::
Register_Conv2D
(
this
);
ops
::
Register_Conv2D
(
this
);
ops
::
Register_DepthwiseConv2d
(
this
);
ops
::
Register_DepthwiseConv2d
(
this
);
ops
::
Register_Eltwise
(
this
);
ops
::
Register_FoldedBatchNorm
(
this
);
ops
::
Register_FoldedBatchNorm
(
this
);
ops
::
Register_FullyConnected
(
this
);
ops
::
Register_FusedConv2D
(
this
);
ops
::
Register_FusedConv2D
(
this
);
ops
::
Register_GlobalAvgPooling
(
this
);
ops
::
Register_GlobalAvgPooling
(
this
);
ops
::
Register_ImageToBuffer
(
this
);
ops
::
Register_ImageToBuffer
(
this
);
ops
::
Register_MatMul
(
this
);
ops
::
Register_Pooling
(
this
);
ops
::
Register_Pooling
(
this
);
ops
::
Register_Proposal
(
this
);
ops
::
Register_PSROIAlign
(
this
);
ops
::
Register_Reshape
(
this
);
ops
::
Register_ResizeBilinear
(
this
);
ops
::
Register_ResizeBilinear
(
this
);
ops
::
Register_Slice
(
this
);
ops
::
Register_Softmax
(
this
);
ops
::
Register_Softmax
(
this
);
ops
::
Register_SpaceToBatchND
(
this
);
ops
::
Register_SpaceToBatchND
(
this
);
ops
::
Register_MatMul
(
this
);
ops
::
Register_WinogradTransform
(
this
);
ops
::
Register_WinogradInverseTransform
(
this
);
ops
::
Register_WinogradInverseTransform
(
this
);
ops
::
Register_Reshape
(
this
);
ops
::
Register_WinogradTransform
(
this
);
ops
::
Register_Eltwise
(
this
);
ops
::
Register_FullyConnected
(
this
);
ops
::
Register_Slice
(
this
);
ops
::
Register_Proposal
(
this
);
ops
::
Register_PSROIAlign
(
this
);
}
}
}
// namespace mace
}
// namespace mace
mace/python/tools/str2vec_maps.cc.tmpl
浏览文件 @
5efffb7c
...
@@ -10,7 +10,7 @@
...
@@ -10,7 +10,7 @@
namespace mace {
namespace mace {
extern const std::map<std::string, std::vector<{{data_type}}>> {{variable_name}}=
extern const std::map<std::string, std::vector<{{data_type}}>> {{variable_name}}
=
{
{
{% for key, value in maps.iteritems() %}
{% for key, value in maps.iteritems() %}
{
{
...
@@ -24,4 +24,4 @@ extern const std::map<std::string, std::vector<{{data_type}}>> {{variable_name}}
...
@@ -24,4 +24,4 @@ extern const std::map<std::string, std::vector<{{data_type}}>> {{variable_name}}
{% endfor %}
{% endfor %}
};
};
} // namespace
} // namespace
mace
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录