Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
Xiaomi
Mace
提交
19084971
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看板
提交
19084971
编写于
3月 29, 2019
作者:
刘
刘琦
浏览文件
操作
浏览文件
下载
差异文件
Merge branch 'fix-df-bug' into 'master'
Fix data format bug in MaceTensor. See merge request !1037
上级
6fde5497
db1ab2cd
变更
3
隐藏空白更改
内联
并排
Showing
3 changed file
with
6 addition
and
4 deletion
+6
-4
.travis.yml
.travis.yml
+1
-1
mace/core/runtime/cpu/cpu_runtime.cc
mace/core/runtime/cpu/cpu_runtime.cc
+1
-0
mace/libmace/mace.cc
mace/libmace/mace.cc
+4
-3
未找到文件。
.travis.yml
浏览文件 @
19084971
...
...
@@ -125,7 +125,7 @@ jobs:
-
python tools/bazel_adb_run.py --target="//mace/test:mace_api_test" --run_target=False --target_abis=arm64-v8a || exit
1
-
python tools/bazel_adb_run.py --target="//mace/test:mace_api_mt_test" --run_target=False --target_abis=arm64-v8a || exit
1
-
echo 'Extra Test on ARM64'
-
python tools/bazel_adb_run.py --target="//mace/utils:
tuner
_test" --run_target=False --target_abis=arm64-v8a || exit
1
-
python tools/bazel_adb_run.py --target="//mace/utils:
utils
_test" --run_target=False --target_abis=arm64-v8a || exit
1
env
:
TYPE=Extra-Test-ARM64-v8a
os
:
linux
dist
:
xenial
...
...
mace/core/runtime/cpu/cpu_runtime.cc
浏览文件 @
19084971
...
...
@@ -20,6 +20,7 @@
#include <algorithm>
#include <cerrno>
#include <cmath>
#include <cstring>
#include <fstream>
#include <string>
...
...
mace/libmace/mace.cc
浏览文件 @
19084971
...
...
@@ -293,9 +293,10 @@ MaceTensor::MaceTensor(const std::vector<int64_t> &shape,
std
::
shared_ptr
<
float
>
data
,
const
DataFormat
format
)
{
MACE_CHECK_NOTNULL
(
data
.
get
());
MACE_CHECK
(
format
==
DataFormat
::
NHWC
||
format
==
DataFormat
::
NCHW
||
format
==
OIHW
,
"MACE only support NHWC, NCHW and OIHW formats of input now."
);
MACE_CHECK
(
format
==
DataFormat
::
DF_NONE
||
format
==
DataFormat
::
NHWC
||
format
==
DataFormat
::
NCHW
||
format
==
OIHW
,
"MACE only support DF_NONE, NHWC, NCHW and OIHW "
"formats of input now."
);
impl_
=
make_unique
<
MaceTensor
::
Impl
>
();
impl_
->
shape
=
shape
;
impl_
->
data
=
data
;
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录