Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
as350144
Mace
提交
acde2afd
Mace
项目概览
as350144
/
Mace
与 Fork 源项目一致
Fork自
Xiaomi / Mace
通知
2
Star
1
Fork
1
代码
文件
提交
分支
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看板
体验新版 GitCode,发现更多精彩内容 >>
提交
acde2afd
编写于
12月 18, 2017
作者:
刘
刘琦
浏览文件
操作
浏览文件
下载
差异文件
Merge branch 'generate-header' into 'master'
Fix memory related bug. See merge request !164
上级
ad26d45d
9fe23eca
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
6 addition
and
4 deletion
+6
-4
mace/python/tools/model.template
mace/python/tools/model.template
+6
-4
未找到文件。
mace/python/tools/model.template
浏览文件 @
acde2afd
...
...
@@ -66,7 +66,6 @@ static void CreateNetArg(mace::NetDef &net_def) {
static void UpdateOp(mace::OperatorDef &op,
const std::string &name,
const std::string &type,
const int mem_id,
const std::vector<std::string> &inputs,
const std::vector<std::string> &outputs,
const std::vector<mace::DataType> &output_types) {
...
...
@@ -74,7 +73,6 @@ static void UpdateOp(mace::OperatorDef &op,
op.set_type(type);
op.set_input(inputs);
op.set_output(outputs);
op.set_mem_id(mem_id);
op.set_output_type(output_types);
}
...
...
@@ -108,11 +106,15 @@ static void CreateOperators(std::vector<mace::OperatorDef> &ops) {
{% endif %}
{% endfor %}
{% if net.op[i].HasField('mem_id') %}
ops[{{i}}].set_mem_id({{net.op[i].mem_id}});
{% endif %}
{% for shape in net.op[i].output_shape %}
ops[{{i}}].add_output_shape(mace::OutputShape({ {{ shape.dims|join(', ') }} }));
{% endfor %}
UpdateOp(ops[{{i}}], {{ net.op[i].name|tojson }}, {{ net.op[i].type|tojson}},
{{ net.op[i].mem_id }},
UpdateOp(ops[{{i}}], {{ net.op[i].name|tojson }}, {{ net.op[i].type|tojson}},
{ {{ net.op[i].input|stringfy }} },
{ {{ net.op[i].output|stringfy }} },
{ {{ net.op[i].output_type|join(', ') }} });
...
...
@@ -139,7 +141,7 @@ static void CreateTensors(std::vector<mace::TensorProto> &tensors) {
{% if net.mem_arena.mem_block|length != 0 %}
static void CreateMemoryArena(mace::MemoryArena &mem_arena) {
auto
mem_block = mem_arena.mutable_mem_block();
std::vector<mace::MemoryBlock> &
mem_block = mem_arena.mutable_mem_block();
mem_block.reserve({{ net.mem_arena.mem_block|length }});
{% for mem_blk in net.mem_arena.mem_block %}
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录