Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
as350144
Mace
提交
cdd4b4c7
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,发现更多精彩内容 >>
提交
cdd4b4c7
编写于
11月 07, 2017
作者:
L
Liangliang He
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Add naive resize bilinear opencl kernel without caching
上级
2b7d72d8
变更
12
隐藏空白更改
内联
并排
Showing
12 changed file
with
185 addition
and
50 deletion
+185
-50
mace/kernels/conv_2d.h
mace/kernels/conv_2d.h
+12
-12
mace/kernels/neon/conv_2d_neon.cc
mace/kernels/neon/conv_2d_neon.cc
+2
-2
mace/kernels/opencl/cl/resize_bilinear.cl
mace/kernels/opencl/cl/resize_bilinear.cl
+36
-0
mace/kernels/opencl/conv_2d_opencl.cc
mace/kernels/opencl/conv_2d_opencl.cc
+2
-2
mace/kernels/opencl/conv_2d_opencl_1x1.cc
mace/kernels/opencl/conv_2d_opencl_1x1.cc
+21
-21
mace/kernels/opencl/conv_2d_opencl_3x3.cc
mace/kernels/opencl/conv_2d_opencl_3x3.cc
+3
-3
mace/kernels/opencl/depthwise_conv_opencl.cc
mace/kernels/opencl/depthwise_conv_opencl.cc
+2
-2
mace/kernels/opencl/resize_bilinear_opencl.cc
mace/kernels/opencl/resize_bilinear_opencl.cc
+46
-2
mace/kernels/resize_bilinear.h
mace/kernels/resize_bilinear.h
+2
-0
mace/ops/resize_bilinear.h
mace/ops/resize_bilinear.h
+0
-1
mace/ops/resize_bilinear_benchmark.cc
mace/ops/resize_bilinear_benchmark.cc
+4
-4
mace/ops/resize_bilinear_test.cc
mace/ops/resize_bilinear_test.cc
+55
-1
未找到文件。
mace/kernels/conv_2d.h
浏览文件 @
cdd4b4c7
...
@@ -27,18 +27,18 @@ struct Conv2dFunctor {
...
@@ -27,18 +27,18 @@ struct Conv2dFunctor {
MACE_CHECK_NOTNULL
(
filter
);
MACE_CHECK_NOTNULL
(
filter
);
MACE_CHECK_NOTNULL
(
output
);
MACE_CHECK_NOTNULL
(
output
);
index_t
batch
=
output
->
shape
()[
0
]
;
index_t
batch
=
output
->
dim
(
0
)
;
index_t
channels
=
output
->
shape
()[
1
]
;
index_t
channels
=
output
->
dim
(
1
)
;
index_t
height
=
output
->
shape
()[
2
]
;
index_t
height
=
output
->
dim
(
2
)
;
index_t
width
=
output
->
shape
()[
3
]
;
index_t
width
=
output
->
dim
(
3
)
;
index_t
input_batch
=
input
->
shape
()[
0
]
;
index_t
input_batch
=
input
->
dim
(
0
)
;
index_t
input_channels
=
input
->
shape
()[
1
]
;
index_t
input_channels
=
input
->
dim
(
1
)
;
index_t
input_height
=
input
->
shape
()[
2
]
;
index_t
input_height
=
input
->
dim
(
2
)
;
index_t
input_width
=
input
->
shape
()[
3
]
;
index_t
input_width
=
input
->
dim
(
3
)
;
index_t
kernel_h
=
filter
->
shape
()[
2
]
;
index_t
kernel_h
=
filter
->
dim
(
2
)
;
index_t
kernel_w
=
filter
->
shape
()[
3
]
;
index_t
kernel_w
=
filter
->
dim
(
3
)
;
int
stride_h
=
strides_
[
0
];
int
stride_h
=
strides_
[
0
];
int
stride_w
=
strides_
[
1
];
int
stride_w
=
strides_
[
1
];
...
...
mace/kernels/neon/conv_2d_neon.cc
浏览文件 @
cdd4b4c7
...
@@ -61,8 +61,8 @@ void Conv2dFunctor<DeviceType::NEON, float>::operator()(const Tensor *input,
...
@@ -61,8 +61,8 @@ void Conv2dFunctor<DeviceType::NEON, float>::operator()(const Tensor *input,
{
nullptr
,
nullptr
},
{
nullptr
,
nullptr
},
{
Conv2dNeonK5x5S1
,
nullptr
}};
{
Conv2dNeonK5x5S1
,
nullptr
}};
// not implement yet
// not implement yet
index_t
kernel_h
=
filter
->
shape
()[
2
]
;
index_t
kernel_h
=
filter
->
dim
(
2
)
;
index_t
kernel_w
=
filter
->
shape
()[
3
]
;
index_t
kernel_w
=
filter
->
dim
(
3
)
;
if
(
kernel_h
!=
kernel_w
||
kernel_h
>
5
||
strides_
[
0
]
!=
strides_
[
1
]
||
if
(
kernel_h
!=
kernel_w
||
kernel_h
>
5
||
strides_
[
0
]
!=
strides_
[
1
]
||
strides_
[
0
]
>
2
||
dilations_
[
0
]
!=
1
||
dilations_
[
1
]
!=
1
||
strides_
[
0
]
>
2
||
dilations_
[
0
]
!=
1
||
dilations_
[
1
]
!=
1
||
selector
[
kernel_h
-
1
][
strides_
[
0
]
-
1
]
==
nullptr
)
{
selector
[
kernel_h
-
1
][
strides_
[
0
]
-
1
]
==
nullptr
)
{
...
...
mace/kernels/opencl/cl/resize_bilinear.cl
0 → 100644
浏览文件 @
cdd4b4c7
__kernel
void
resize_bilinear_nocache
(
__global
const
float
*input,
/*
n
*
c,
h,
w
*/
__global
float
*output
/*
n
*
c,
h,
w
*/,
__private
const
float
height_scale,
__private
const
float
width_scale,
__private
const
int
in_height,
__private
const
int
in_width
)
{
const
int
c
=
get_global_id
(
0
)
;
const
int
h
=
get_global_id
(
1
)
;
const
int
w
=
get_global_id
(
2
)
;
const
int
channels
=
get_global_size
(
0
)
;
const
int
height
=
get_global_size
(
1
)
;
const
int
width
=
get_global_size
(
2
)
;
const
float
h_in
=
h
*
height_scale
;
const
float
w_in
=
w
*
width_scale
;
const
int
h_lower
=
max
(
0
,
(
int
)
floor
(
h_in
))
;
const
int
h_upper
=
min
(
in_height
-
1
,
h_lower
+
1
)
;
const
int
w_lower
=
max
(
0
,
(
int
)
floor
(
w_in
))
;
const
int
w_upper
=
min
(
in_width
-
1
,
w_lower
+
1
)
;
const
float
h_lerp
=
h_in
-
h_lower
;
const
float
w_lerp
=
w_in
-
w_lower
;
const
float
*input_base
=
input
+
c
*
in_height
*
in_width
;
float
*output_base
=
output
+
c
*
height
*
width
;
float
top_left
=
input_base[h_lower
*
in_width
+
w_lower]
;
float
top_right
=
input_base[h_lower
*
in_width
+
w_upper]
;
float
bottom_left
=
input_base[h_upper
*
in_width
+
w_lower]
;
float
bottom_right
=
input_base[h_upper
*
in_width
+
w_upper]
;
const
float
top
=
top_left
+
(
top_right
-
top_left
)
*
w_lerp
;
const
float
bottom
=
bottom_left
+
(
bottom_right
-
bottom_left
)
*
w_lerp
;
output_base[h
*
width
+
w]
=
top
+
(
bottom
-
top
)
*
h_lerp
;
}
mace/kernels/opencl/conv_2d_opencl.cc
浏览文件 @
cdd4b4c7
...
@@ -30,8 +30,8 @@ void Conv2dFunctor<DeviceType::OPENCL, float>::operator()(const Tensor *input,
...
@@ -30,8 +30,8 @@ void Conv2dFunctor<DeviceType::OPENCL, float>::operator()(const Tensor *input,
{
nullptr
,
nullptr
},
{
nullptr
,
nullptr
},
{
nullptr
,
nullptr
}};
{
nullptr
,
nullptr
}};
index_t
kernel_h
=
filter
->
shape
()[
2
]
;
index_t
kernel_h
=
filter
->
dim
(
2
)
;
index_t
kernel_w
=
filter
->
shape
()[
3
]
;
index_t
kernel_w
=
filter
->
dim
(
3
)
;
if
(
kernel_h
!=
kernel_w
||
kernel_h
>
5
||
strides_
[
0
]
!=
strides_
[
1
]
||
if
(
kernel_h
!=
kernel_w
||
kernel_h
>
5
||
strides_
[
0
]
!=
strides_
[
1
]
||
strides_
[
0
]
>
2
||
dilations_
[
0
]
!=
1
||
dilations_
[
1
]
!=
1
||
strides_
[
0
]
>
2
||
dilations_
[
0
]
!=
1
||
dilations_
[
1
]
!=
1
||
selector
[
kernel_h
-
1
][
strides_
[
0
]
-
1
]
==
nullptr
)
{
selector
[
kernel_h
-
1
][
strides_
[
0
]
-
1
]
==
nullptr
)
{
...
...
mace/kernels/opencl/conv_2d_opencl_1x1.cc
浏览文件 @
cdd4b4c7
...
@@ -15,11 +15,11 @@ void Conv1x1Naive(const Tensor *input,
...
@@ -15,11 +15,11 @@ void Conv1x1Naive(const Tensor *input,
const
Tensor
*
filter
,
const
Tensor
*
filter
,
const
Tensor
*
bias
,
const
Tensor
*
bias
,
Tensor
*
output
)
{
Tensor
*
output
)
{
const
index_t
batch
=
output
->
shape
()[
0
]
;
const
index_t
batch
=
output
->
dim
(
0
)
;
const
index_t
channels
=
output
->
shape
()[
1
]
;
const
index_t
channels
=
output
->
dim
(
1
)
;
const
index_t
height
=
output
->
shape
()[
2
]
;
const
index_t
height
=
output
->
dim
(
2
)
;
const
index_t
width
=
output
->
shape
()[
3
]
;
const
index_t
width
=
output
->
dim
(
3
)
;
const
index_t
input_channels
=
input
->
shape
()[
1
]
;
const
index_t
input_channels
=
input
->
dim
(
1
)
;
auto
runtime
=
OpenCLRuntime
::
Get
();
auto
runtime
=
OpenCLRuntime
::
Get
();
auto
program
=
runtime
->
program
();
auto
program
=
runtime
->
program
();
...
@@ -46,11 +46,11 @@ void Conv1x1V2(const Tensor *input,
...
@@ -46,11 +46,11 @@ void Conv1x1V2(const Tensor *input,
const
Tensor
*
filter
,
const
Tensor
*
filter
,
const
Tensor
*
bias
,
const
Tensor
*
bias
,
Tensor
*
output
)
{
Tensor
*
output
)
{
const
index_t
batch
=
output
->
shape
()[
0
]
;
const
index_t
batch
=
output
->
dim
(
0
)
;
const
index_t
channels
=
output
->
shape
()[
1
]
;
const
index_t
channels
=
output
->
dim
(
1
)
;
const
index_t
height
=
output
->
shape
()[
2
]
;
const
index_t
height
=
output
->
dim
(
2
)
;
const
index_t
width
=
output
->
shape
()[
3
]
;
const
index_t
width
=
output
->
dim
(
3
)
;
const
index_t
input_channels
=
input
->
shape
()[
1
]
;
const
index_t
input_channels
=
input
->
dim
(
1
)
;
auto
runtime
=
OpenCLRuntime
::
Get
();
auto
runtime
=
OpenCLRuntime
::
Get
();
auto
program
=
runtime
->
program
();
auto
program
=
runtime
->
program
();
...
@@ -88,11 +88,11 @@ void Conv1x1V3(const Tensor *input,
...
@@ -88,11 +88,11 @@ void Conv1x1V3(const Tensor *input,
const
Tensor
*
filter
,
const
Tensor
*
filter
,
const
Tensor
*
bias
,
const
Tensor
*
bias
,
Tensor
*
output
)
{
Tensor
*
output
)
{
const
index_t
batch
=
output
->
shape
()[
0
]
;
const
index_t
batch
=
output
->
dim
(
0
)
;
const
index_t
channels
=
output
->
shape
()[
1
]
;
const
index_t
channels
=
output
->
dim
(
1
)
;
const
index_t
height
=
output
->
shape
()[
2
]
;
const
index_t
height
=
output
->
dim
(
2
)
;
const
index_t
width
=
output
->
shape
()[
3
]
;
const
index_t
width
=
output
->
dim
(
3
)
;
const
index_t
input_channels
=
input
->
shape
()[
1
]
;
const
index_t
input_channels
=
input
->
dim
(
1
)
;
auto
runtime
=
OpenCLRuntime
::
Get
();
auto
runtime
=
OpenCLRuntime
::
Get
();
auto
program
=
runtime
->
program
();
auto
program
=
runtime
->
program
();
...
@@ -174,13 +174,13 @@ extern void Conv2dOpenclK1x1S1(const Tensor *input,
...
@@ -174,13 +174,13 @@ extern void Conv2dOpenclK1x1S1(const Tensor *input,
const
Tensor
*
filter
,
const
Tensor
*
filter
,
const
Tensor
*
bias
,
const
Tensor
*
bias
,
Tensor
*
output
)
{
Tensor
*
output
)
{
const
index_t
batch
=
output
->
shape
()[
0
]
;
const
index_t
batch
=
output
->
dim
(
0
)
;
const
index_t
height
=
output
->
shape
()[
2
]
;
const
index_t
height
=
output
->
dim
(
2
)
;
const
index_t
width
=
output
->
shape
()[
3
]
;
const
index_t
width
=
output
->
dim
(
3
)
;
const
index_t
input_batch
=
input
->
shape
()[
0
]
;
const
index_t
input_batch
=
input
->
dim
(
0
)
;
const
index_t
input_height
=
input
->
shape
()[
2
]
;
const
index_t
input_height
=
input
->
dim
(
2
)
;
const
index_t
input_width
=
input
->
shape
()[
3
]
;
const
index_t
input_width
=
input
->
dim
(
3
)
;
MACE_CHECK
(
input_batch
==
batch
&&
input_height
==
height
&&
MACE_CHECK
(
input_batch
==
batch
&&
input_height
==
height
&&
input_width
==
width
);
input_width
==
width
);
...
...
mace/kernels/opencl/conv_2d_opencl_3x3.cc
浏览文件 @
cdd4b4c7
...
@@ -11,9 +11,9 @@ namespace kernels {
...
@@ -11,9 +11,9 @@ namespace kernels {
static
void
InnerConv2dK3x3S12
(
const
Tensor
*
input
,
const
Tensor
*
filter
,
static
void
InnerConv2dK3x3S12
(
const
Tensor
*
input
,
const
Tensor
*
filter
,
const
Tensor
*
bias
,
const
uint32_t
stride
,
Tensor
*
output
)
{
const
Tensor
*
bias
,
const
uint32_t
stride
,
Tensor
*
output
)
{
const
index_t
channels
=
output
->
shape
()[
1
]
;
const
index_t
channels
=
output
->
dim
(
1
)
;
const
index_t
height
=
output
->
shape
()[
2
]
;
const
index_t
height
=
output
->
dim
(
2
)
;
const
index_t
width
=
output
->
shape
()[
3
]
;
const
index_t
width
=
output
->
dim
(
3
)
;
MACE_CHECK
(
input
->
dim
(
0
)
==
output
->
dim
(
0
));
MACE_CHECK
(
input
->
dim
(
0
)
==
output
->
dim
(
0
));
...
...
mace/kernels/opencl/depthwise_conv_opencl.cc
浏览文件 @
cdd4b4c7
...
@@ -27,8 +27,8 @@ void DepthwiseConv2dFunctor<DeviceType::OPENCL, float>::operator()(const Tensor
...
@@ -27,8 +27,8 @@ void DepthwiseConv2dFunctor<DeviceType::OPENCL, float>::operator()(const Tensor
{
nullptr
,
nullptr
},
{
nullptr
,
nullptr
},
{
nullptr
,
nullptr
}};
{
nullptr
,
nullptr
}};
index_t
kernel_h
=
filter
->
shape
()[
2
]
;
index_t
kernel_h
=
filter
->
dim
(
2
)
;
index_t
kernel_w
=
filter
->
shape
()[
3
]
;
index_t
kernel_w
=
filter
->
dim
(
3
)
;
if
(
kernel_h
!=
kernel_w
||
kernel_h
>
5
||
strides_
[
0
]
!=
strides_
[
1
]
||
if
(
kernel_h
!=
kernel_w
||
kernel_h
>
5
||
strides_
[
0
]
!=
strides_
[
1
]
||
strides_
[
0
]
>
2
||
dilations_
[
0
]
!=
1
||
dilations_
[
1
]
!=
1
||
strides_
[
0
]
>
2
||
dilations_
[
0
]
!=
1
||
dilations_
[
1
]
!=
1
||
selector
[
kernel_h
-
1
][
strides_
[
0
]
-
1
]
==
nullptr
)
{
selector
[
kernel_h
-
1
][
strides_
[
0
]
-
1
]
==
nullptr
)
{
...
...
mace/kernels/opencl/resize_bilinear_opencl.cc
浏览文件 @
cdd4b4c7
...
@@ -2,15 +2,59 @@
...
@@ -2,15 +2,59 @@
// Copyright (c) 2017 XiaoMi All rights reserved.
// Copyright (c) 2017 XiaoMi All rights reserved.
//
//
#include "mace/
kernels/resize_bilinear
.h"
#include "mace/
core/runtime/opencl/opencl_runtime
.h"
#include "mace/core/tensor.h"
#include "mace/core/tensor.h"
#include "mace/kernels/resize_bilinear.h"
namespace
mace
{
namespace
mace
{
namespace
kernels
{
namespace
kernels
{
template
<
>
template
<
>
void
ResizeBilinearFunctor
<
DeviceType
::
OPENCL
,
float
>::
operator
()(
void
ResizeBilinearFunctor
<
DeviceType
::
OPENCL
,
float
>::
operator
()(
const
Tensor
*
input
,
const
Tensor
*
resize_dims
,
Tensor
*
output
)
{}
const
Tensor
*
input
,
const
Tensor
*
resize_dims
,
Tensor
*
output
)
{
const
index_t
batch
=
input
->
dim
(
0
);
const
index_t
channels
=
input
->
dim
(
1
);
const
index_t
in_height
=
input
->
dim
(
2
);
const
index_t
in_width
=
input
->
dim
(
3
);
index_t
out_height
;
index_t
out_width
;
{
MACE_CHECK
(
resize_dims
->
dim_size
()
==
1
);
Tensor
::
MappingGuard
resize_dims_mapper
(
resize_dims
);
auto
dims_data
=
resize_dims
->
data
<
index_t
>
();
out_height
=
dims_data
[
0
];
out_width
=
dims_data
[
1
];
}
std
::
vector
<
index_t
>
out_shape
{
batch
,
channels
,
out_height
,
out_width
};
output
->
Resize
(
out_shape
);
float
height_scale
=
CalculateResizeScale
(
in_height
,
out_height
,
align_corners_
);
float
width_scale
=
CalculateResizeScale
(
in_width
,
out_width
,
align_corners_
);
auto
runtime
=
OpenCLRuntime
::
Get
();
auto
program
=
runtime
->
program
();
auto
rb_kernel
=
cl
::
Kernel
(
program
,
"resize_bilinear_nocache"
);
const
uint32_t
kwg_size
=
runtime
->
GetKernelMaxWorkGroupSize
(
rb_kernel
);
uint32_t
idx
=
0
;
rb_kernel
.
setArg
(
idx
++
,
*
(
static_cast
<
const
cl
::
Buffer
*>
(
input
->
buffer
())));
rb_kernel
.
setArg
(
idx
++
,
*
(
static_cast
<
cl
::
Buffer
*>
(
output
->
buffer
())));
rb_kernel
.
setArg
(
idx
++
,
static_cast
<
float
>
(
height_scale
));
rb_kernel
.
setArg
(
idx
++
,
static_cast
<
float
>
(
width_scale
));
rb_kernel
.
setArg
(
idx
++
,
static_cast
<
int
>
(
in_height
));
rb_kernel
.
setArg
(
idx
++
,
static_cast
<
int
>
(
in_width
));
auto
command_queue
=
runtime
->
command_queue
();
cl_int
error
=
command_queue
.
enqueueNDRangeKernel
(
rb_kernel
,
cl
::
NullRange
,
cl
::
NDRange
(
static_cast
<
int
>
(
batch
*
channels
),
static_cast
<
int
>
(
out_height
),
static_cast
<
int
>
(
out_width
)),
cl
::
NDRange
(
1
,
16
,
kwg_size
/
16
));
MACE_CHECK
(
error
==
CL_SUCCESS
,
error
);
}
}
// namespace kernels
}
// namespace kernels
}
// namespace mace
}
// namespace mace
mace/kernels/resize_bilinear.h
浏览文件 @
cdd4b4c7
...
@@ -127,6 +127,8 @@ struct ResizeBilinearFunctor {
...
@@ -127,6 +127,8 @@ struct ResizeBilinearFunctor {
vector
<
index_t
>
out_shape
{
n
,
channels
,
out_height
,
out_width
};
vector
<
index_t
>
out_shape
{
n
,
channels
,
out_height
,
out_width
};
output
->
Resize
(
out_shape
);
output
->
Resize
(
out_shape
);
Tensor
::
MappingGuard
input_mapper
(
input
);
Tensor
::
MappingGuard
output_mapper
(
output
);
const
T
*
input_data
=
input
->
data
<
T
>
();
const
T
*
input_data
=
input
->
data
<
T
>
();
T
*
output_data
=
output
->
mutable_data
<
T
>
();
T
*
output_data
=
output
->
mutable_data
<
T
>
();
...
...
mace/ops/resize_bilinear.h
浏览文件 @
cdd4b4c7
...
@@ -28,7 +28,6 @@ class ResizeBilinearOp : public Operator<D, T> {
...
@@ -28,7 +28,6 @@ class ResizeBilinearOp : public Operator<D, T> {
MACE_CHECK
(
resize_dims
->
dim_size
()
==
1
,
MACE_CHECK
(
resize_dims
->
dim_size
()
==
1
,
"resize dim must be 2-dimensional."
,
resize_dims
->
dim_size
());
"resize dim must be 2-dimensional."
,
resize_dims
->
dim_size
());
functor_
(
input
,
resize_dims
,
output
);
functor_
(
input
,
resize_dims
,
output
);
return
true
;
return
true
;
}
}
...
...
mace/ops/resize_bilinear_benchmark.cc
浏览文件 @
cdd4b4c7
...
@@ -26,10 +26,10 @@ static void ResizeBilinearBenchmark(int iters,
...
@@ -26,10 +26,10 @@ static void ResizeBilinearBenchmark(int iters,
.
Finalize
(
net
.
NewOperatorDef
());
.
Finalize
(
net
.
NewOperatorDef
());
// Add input data
// Add input data
net
.
AddRandomInput
<
D
eviceType
::
CPU
,
float
>
(
net
.
AddRandomInput
<
D
,
float
>
(
"Input"
,
"Input"
,
{
batch
,
channels
,
input_height
,
input_width
});
{
batch
,
channels
,
input_height
,
input_width
});
net
.
AddInputFromArray
<
D
eviceType
::
CPU
,
index_t
>
(
net
.
AddInputFromArray
<
D
,
index_t
>
(
"OutSize"
,
{
2
},
"OutSize"
,
{
2
},
{
output_height
,
output_width
});
{
output_height
,
output_width
});
// Warm-up
// Warm-up
for
(
int
i
=
0
;
i
<
5
;
++
i
)
{
for
(
int
i
=
0
;
i
<
5
;
++
i
)
{
...
...
mace/ops/resize_bilinear_test.cc
浏览文件 @
cdd4b4c7
...
@@ -10,7 +10,7 @@ using namespace mace;
...
@@ -10,7 +10,7 @@ using namespace mace;
class
ResizeBilinearTest
:
public
OpsTestBase
{};
class
ResizeBilinearTest
:
public
OpsTestBase
{};
TEST_F
(
ResizeBilinearTest
,
ResizeBilinearWOAlignCorners
)
{
TEST_F
(
ResizeBilinearTest
,
CPU
ResizeBilinearWOAlignCorners
)
{
testing
::
internal
::
LogToStderr
();
testing
::
internal
::
LogToStderr
();
// Construct graph
// Construct graph
auto
&
net
=
test_net
();
auto
&
net
=
test_net
();
...
@@ -60,3 +60,57 @@ TEST_F(ResizeBilinearTest, ResizeBilinearWAlignCorners) {
...
@@ -60,3 +60,57 @@ TEST_F(ResizeBilinearTest, ResizeBilinearWAlignCorners) {
ExpectTensorNear
<
float
>
(
*
expected
,
*
net
.
GetOutput
(
"Output"
),
0.001
);
ExpectTensorNear
<
float
>
(
*
expected
,
*
net
.
GetOutput
(
"Output"
),
0.001
);
}
}
template
<
DeviceType
D
>
void
TestRandomResizeBilinear
()
{
srand
(
time
(
nullptr
));
testing
::
internal
::
LogToStderr
();
for
(
int
round
=
0
;
round
<
10
;
++
round
)
{
index_t
batch
=
1
+
rand
()
%
5
;
index_t
channels
=
1
+
rand
()
%
100
;
index_t
height
=
1
+
rand
()
%
100
;
index_t
width
=
1
+
rand
()
%
100
;
index_t
in_height
=
1
+
rand
()
%
100
;
index_t
in_width
=
1
+
rand
()
%
100
;
// Construct graph
OpsTestNet
net
;
OpDefBuilder
(
"ResizeBilinear"
,
"ResizeBilinearTest"
)
.
Input
(
"Input"
)
.
Input
(
"OutSize"
)
.
Output
(
"Output"
)
.
AddIntArg
(
"align_corners"
,
1
)
.
Finalize
(
net
.
NewOperatorDef
());
// Add input data
net
.
AddRandomInput
<
D
,
float
>
(
"Input"
,
{
batch
,
channels
,
in_height
,
in_width
});
net
.
AddInputFromArray
<
D
,
index_t
>
(
"OutSize"
,
{
2
},
{
height
,
width
});
/*
vector<float> input(24);
std::iota(begin(input), end(input), 0);
net.AddInputFromArray<DeviceType::OPENCL, float>("Input", {1, 3, 2, 4}, input);
net.AddInputFromArray<DeviceType::OPENCL, index_t>("OutSize", {2}, {1, 2});
*/
// Run
net
.
RunOp
(
D
);
Tensor
actual
;
actual
.
Copy
(
*
net
.
GetOutput
(
"Output"
));
// Run on CPU
net
.
RunOp
(
DeviceType
::
CPU
);
Tensor
*
expected
=
net
.
GetOutput
(
"Output"
);
// Check
ExpectTensorNear
<
float
>
(
*
expected
,
actual
,
0.001
);
}
}
TEST_F
(
ResizeBilinearTest
,
NEONRandomResizeBilinear
)
{
TestRandomResizeBilinear
<
DeviceType
::
NEON
>
();
}
TEST_F
(
ResizeBilinearTest
,
OPENCLRandomResizeBilinear
)
{
TestRandomResizeBilinear
<
DeviceType
::
OPENCL
>
();
}
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录