Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
Xiaomi
Mace
提交
8cf29eb7
Mace
项目概览
Xiaomi
/
Mace
通知
107
Star
40
Fork
27
代码
文件
提交
分支
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看板
提交
8cf29eb7
编写于
11月 04, 2019
作者:
L
Liangliang He
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Force jinja escape quoted string to prevent security issues when using third party models
上级
5a0a017c
变更
5
隐藏空白更改
内联
并排
Showing
5 changed file
with
9 addition
and
6 deletion
+9
-6
mace/python/tools/operator.jinja2
mace/python/tools/operator.jinja2
+2
-1
mace/python/tools/str2vec_maps.cc.jinja2
mace/python/tools/str2vec_maps.cc.jinja2
+2
-2
tools/python/template/model.jinja2
tools/python/template/model.jinja2
+1
-1
tools/python/template/operator.jinja2
tools/python/template/operator.jinja2
+2
-1
tools/python/template/tensor_source.jinja2
tools/python/template/tensor_source.jinja2
+2
-1
未找到文件。
mace/python/tools/operator.jinja2
浏览文件 @
8cf29eb7
...
@@ -64,7 +64,8 @@ namespace {{tag}} {
...
@@ -64,7 +64,8 @@ namespace {{tag}} {
{% for i in range(start, end) %}
{% for i in range(start, end) %}
void CreateOperator{{i}}(mace::OperatorDef *op) {
void CreateOperator{{i}}(mace::OperatorDef *op) {
MACE_LATENCY_LOGGER(2, "Create operator {{ net.op[i].name }}");
MACE_LATENCY_LOGGER(2, "Create operator",
{{ net.op[i].name|default("undefined")|tojson }});
mace::Argument *arg = nullptr;
mace::Argument *arg = nullptr;
op->mutable_arg()->Reserve({{ net.op[i].arg|length }});
op->mutable_arg()->Reserve({{ net.op[i].arg|length }});
...
...
mace/python/tools/str2vec_maps.cc.jinja2
浏览文件 @
8cf29eb7
...
@@ -25,10 +25,10 @@ namespace {{tag}} {
...
@@ -25,10 +25,10 @@ namespace {{tag}} {
const std::map<std::string, ClProgramInfo> {{variable_name}} = {
const std::map<std::string, ClProgramInfo> {{variable_name}} = {
{% for key, encrypted_code in maps.items() %}
{% for key, encrypted_code in maps.items() %}
{
{
"{{key}}"
, {
{{key|tojson}}
, {
{
{
{%- for header in encrypted_code['headers'] -%}
{%- for header in encrypted_code['headers'] -%}
"{{header}}"
,
{{header|tojson}}
,
{%- endfor -%}
{%- endfor -%}
},
},
{
{
...
...
tools/python/template/model.jinja2
浏览文件 @
8cf29eb7
...
@@ -134,7 +134,7 @@ void CreateTensors(NetDef *net_def) {
...
@@ -134,7 +134,7 @@ void CreateTensors(NetDef *net_def) {
namespace {{tag}} {
namespace {{tag}} {
MACE_API const std::shared_ptr<NetDef> CreateNet() {
MACE_API const std::shared_ptr<NetDef> CreateNet() {
MACE_LATENCY_LOGGER(1, "Create net
{{ net.name }}"
);
MACE_LATENCY_LOGGER(1, "Create net
", {{ net.name|default("undefined")|tojson }}
);
std::shared_ptr<NetDef> net_def(new NetDef());
std::shared_ptr<NetDef> net_def(new NetDef());
...
...
tools/python/template/operator.jinja2
浏览文件 @
8cf29eb7
...
@@ -64,7 +64,8 @@ namespace {{tag}} {
...
@@ -64,7 +64,8 @@ namespace {{tag}} {
{% for i in range(start, end) %}
{% for i in range(start, end) %}
void CreateOperator{{i}}(mace::OperatorDef *op) {
void CreateOperator{{i}}(mace::OperatorDef *op) {
MACE_LATENCY_LOGGER(2, "Create operator {{ net.op[i].name }}");
MACE_LATENCY_LOGGER(2, "Create operator ",
{{ net.op[i].name|default("undefined")|tojson }});
mace::Argument *arg = nullptr;
mace::Argument *arg = nullptr;
op->mutable_arg()->Reserve({{ net.op[i].arg|length }});
op->mutable_arg()->Reserve({{ net.op[i].arg|length }});
...
...
tools/python/template/tensor_source.jinja2
浏览文件 @
8cf29eb7
...
@@ -23,7 +23,8 @@ namespace mace {
...
@@ -23,7 +23,8 @@ namespace mace {
namespace {{tag}} {
namespace {{tag}} {
void CreateTensor{{tensor_id}}(mace::ConstTensor *const_tensor) {
void CreateTensor{{tensor_id}}(mace::ConstTensor *const_tensor) {
MACE_LATENCY_LOGGER(2, "Create tensor {{ tensor.name }}");
MACE_LATENCY_LOGGER(2, "Create tensor ",
{{ tensor.name|default("undefined")|tojson }});
const_tensor->set_name({{ tensor.name|tojson }});
const_tensor->set_name({{ tensor.name|tojson }});
const_tensor->set_offset({{ tensor.offset }});
const_tensor->set_offset({{ tensor.offset }});
const_tensor->set_data_size({{ tensor.data_size }});
const_tensor->set_data_size({{ tensor.data_size }});
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录