Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
冰之2023
Mace
提交
047d7f42
Mace
项目概览
冰之2023
/
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,体验更适合开发者的 AI 搜索 >>
提交
047d7f42
编写于
11月 27, 2017
作者:
L
liuqi
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Fix some typo and remove unsed code.
上级
5f5fa20b
变更
5
隐藏空白更改
内联
并排
Showing
5 changed file
with
7 addition
and
5 deletion
+7
-5
mace/kernels/opencl/buffer_to_image.cc
mace/kernels/opencl/buffer_to_image.cc
+1
-1
mace/kernels/opencl/conv_2d_opencl_3x3.cc
mace/kernels/opencl/conv_2d_opencl_3x3.cc
+1
-1
mace/kernels/opencl/helper.cc
mace/kernels/opencl/helper.cc
+2
-2
mace/kernels/opencl/helper.h
mace/kernels/opencl/helper.h
+1
-1
mace/ops/conv_2d_test.cc
mace/ops/conv_2d_test.cc
+2
-0
未找到文件。
mace/kernels/opencl/buffer_to_image.cc
浏览文件 @
047d7f42
...
...
@@ -25,7 +25,7 @@ void BufferToImageFunctor<DeviceType::OPENCL, T>::operator()(Tensor *buffer,
std
::
set
<
std
::
string
>
built_options
;
built_options
.
emplace
(
"-DDATA_TYPE="
+
DataTypeToCLType
(
image
->
dtype
()));
built_options
.
emplace
(
"-DCMD_DATA_TYPE="
+
DataTypeToO
pencl
CMDDataType
(
image
->
dtype
()));
built_options
.
emplace
(
"-DCMD_DATA_TYPE="
+
DataTypeToO
PENCL
CMDDataType
(
image
->
dtype
()));
auto
runtime
=
OpenCLRuntime
::
Get
();
string
kernel_name
;
switch
(
type
)
{
...
...
mace/kernels/opencl/conv_2d_opencl_3x3.cc
浏览文件 @
047d7f42
...
...
@@ -26,7 +26,7 @@ static void Conv2d3x3S12(const Tensor *input, const Tensor *filter,
std
::
set
<
std
::
string
>
built_options
;
built_options
.
emplace
(
"-DDATA_TYPE="
+
DataTypeToCLType
(
input
->
dtype
()));
built_options
.
emplace
(
"-DCMD_DATA_TYPE="
+
DataTypeToO
pencl
CMDDataType
(
input
->
dtype
()));
built_options
.
emplace
(
"-DCMD_DATA_TYPE="
+
DataTypeToO
PENCL
CMDDataType
(
input
->
dtype
()));
built_options
.
emplace
(
bias
!=
nullptr
?
"-DBIAS"
:
""
);
auto
runtime
=
OpenCLRuntime
::
Get
();
...
...
mace/kernels/opencl/helper.cc
浏览文件 @
047d7f42
...
...
@@ -17,7 +17,7 @@ void CalInOutputImageShape(const std::vector<index_t> &shape, /* NHWC */
image_shape
[
1
]
=
shape
[
0
]
*
shape
[
1
];
}
// [H * W *
4
, (Oc + 3) / 4]
// [H * W *
RoundUp<4>(Ic)
, (Oc + 3) / 4]
void
CalFilterImageShape
(
const
std
::
vector
<
index_t
>
&
shape
,
/* HWIO*/
std
::
vector
<
size_t
>
&
image_shape
)
{
MACE_CHECK
(
shape
.
size
()
==
4
);
...
...
@@ -82,7 +82,7 @@ std::string DataTypeToCLType(const DataType dt) {
}
}
std
::
string
DataTypeToO
pencl
CMDDataType
(
const
DataType
dt
)
{
std
::
string
DataTypeToO
PENCL
CMDDataType
(
const
DataType
dt
)
{
switch
(
dt
)
{
case
DT_FLOAT
:
return
"f"
;
...
...
mace/kernels/opencl/helper.h
浏览文件 @
047d7f42
...
...
@@ -19,7 +19,7 @@ void CalImage2DShape(const std::vector<index_t> &shape, /* NHWC */
const
BufferType
type
,
std
::
vector
<
size_t
>
&
image_shape
);
std
::
string
DataTypeToO
pencl
CMDDataType
(
const
DataType
dt
);
std
::
string
DataTypeToO
PENCL
CMDDataType
(
const
DataType
dt
);
std
::
string
DataTypeToCLType
(
const
DataType
dt
);
...
...
mace/ops/conv_2d_test.cc
浏览文件 @
047d7f42
...
...
@@ -467,6 +467,7 @@ static void TestComplexConvNxNS12(const std::vector<index_t> &shape) {
OpDefBuilder
(
"Conv2D"
,
"Conv2dTest"
)
.
Input
(
"Input"
)
.
Input
(
"Filter"
)
.
Input
(
"Bias"
)
.
Output
(
"Output"
)
.
AddIntsArg
(
"strides"
,
{
stride_h
,
stride_w
})
.
AddIntArg
(
"padding"
,
type
)
...
...
@@ -493,6 +494,7 @@ static void TestComplexConvNxNS12(const std::vector<index_t> &shape) {
OpDefBuilder
(
"Conv2D"
,
"Conv2dTest"
)
.
Input
(
"InputImage"
)
.
Input
(
"FilterImage"
)
.
Input
(
"BiasImage"
)
.
Output
(
"OutputImage"
)
.
AddIntsArg
(
"strides"
,
{
stride_h
,
stride_w
})
.
AddIntArg
(
"padding"
,
type
)
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录