Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
项目经理老王
Mace
提交
019174a6
Mace
项目概览
项目经理老王
/
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看板
体验新版 GitCode,发现更多精彩内容 >>
提交
019174a6
编写于
4月 17, 2019
作者:
L
liyin
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Update quantize libs to support thread pool
上级
98bb8555
变更
4
显示空白变更内容
内联
并排
Showing
4 changed file
with
10 addition
and
7 deletion
+10
-7
WORKSPACE
WORKSPACE
+6
-6
mace/core/runtime/cpu/cpu_runtime.h
mace/core/runtime/cpu/cpu_runtime.h
+1
-1
mace/ops/depthwise_conv2d.cc
mace/ops/depthwise_conv2d.cc
+1
-0
tools/bazel.rc
tools/bazel.rc
+2
-0
未找到文件。
WORKSPACE
浏览文件 @
019174a6
...
...
@@ -79,19 +79,19 @@ new_http_archive(
http_archive
(
name
=
"gemmlowp"
,
sha256
=
"
afbea037aee2d21b625985238486b4219396f9c2550b0fde3157fab4d2580205
"
,
strip_prefix
=
"gemmlowp-master-
1f6d8d442805a400c74e63a4a017390733df2e28
"
,
sha256
=
"
f340384e7728cea605e83597593699dfe8d13ff333b834d24c256935e3dc1758
"
,
strip_prefix
=
"gemmlowp-master-
48c0547a046d49b466aa01e3a82a18028f288924
"
,
urls
=
[
"http://cnbj1.fds.api.xiaomi.com/mace/third-party/gemmlowp/gemmlowp-master-
1f6d8d442805a400c74e63a4a017390733df2e28
.zip"
,
"http://cnbj1.fds.api.xiaomi.com/mace/third-party/gemmlowp/gemmlowp-master-
48c0547a046d49b466aa01e3a82a18028f288924
.zip"
,
],
)
http_archive
(
name
=
"tflite"
,
sha256
=
"
8b4c1b2ad2d31da9859e17b0ad551b12e1db7ff2faf7e83218901ab48d9fa91a
"
,
strip_prefix
=
"tensorflow-mace-d
fabaf85145e4d5ad39f34a0cea57b44c32dbe43
"
,
sha256
=
"
6f2671a02fe635a82c289c8c40a6e5bc24670ff1d4c3c2ab4a7aa9b825256a18
"
,
strip_prefix
=
"tensorflow-mace-d
73e88fc830320d3818ac24e57cd441820a85cc9
"
,
urls
=
[
"http://cnbj1.fds.api.xiaomi.com/mace/third-party/tflite/tensorflow-mace-d
fabaf85145e4d5ad39f34a0cea57b44c32dbe43
.zip"
,
"http://cnbj1.fds.api.xiaomi.com/mace/third-party/tflite/tensorflow-mace-d
73e88fc830320d3818ac24e57cd441820a85cc9
.zip"
,
],
)
...
...
mace/core/runtime/cpu/cpu_runtime.h
浏览文件 @
019174a6
...
...
@@ -57,7 +57,7 @@ class CPURuntime {
gemmlowp
::
GemmContext
*
GetGemmlowpContext
()
{
if
(
gemm_context_
==
nullptr
)
{
gemm_context_
=
new
gemmlowp
::
GemmContext
();
gemm_context_
=
new
gemmlowp
::
GemmContext
(
thread_pool_
);
}
return
static_cast
<
gemmlowp
::
GemmContext
*>
(
gemm_context_
);
}
...
...
mace/ops/depthwise_conv2d.cc
浏览文件 @
019174a6
...
...
@@ -250,6 +250,7 @@ class DepthwiseConv2dOp<DeviceType::CPU, uint8_t>
std
::
vector
<
index_t
>
bias_shape
{
out_channels
};
tflite
::
optimized_ops
::
DepthwiseConv
(
&
context
->
device
()
->
cpu_runtime
()
->
thread_pool
(),
input_data
,
ShapeToTfliteDims
(
input
->
shape
()),
-
input
->
zero_point
(),
filter_data
,
ShapeToTfliteDims
(
filter_shape
),
-
filter
->
zero_point
(),
bias_data
,
ShapeToTfliteDims
(
bias_shape
),
stride_w
,
stride_h
,
...
...
tools/bazel.rc
浏览文件 @
019174a6
...
...
@@ -8,6 +8,8 @@ build --copt=-std=c++11
build --copt=-fPIC
build --copt=-D_GLIBCXX_USE_C99_MATH_TR1
build --copt=-DMACE_OBFUSCATE_LITERALS
build --copt=-DGEMMLOWP_USE_MACE_THREAD_POOL
build --copt=-DMACE_DEPTHWISE_U8_USE_MULTI_THREAD
# Usage example: bazel build --config android
build:android --linkopt=-pie
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录