Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
Mr.Vain
Mace
提交
d804bca8
Mace
项目概览
Mr.Vain
/
Mace
与 Fork 源项目一致
Fork自
Xiaomi / Mace
通知
1
Star
1
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 搜索 >>
提交
d804bca8
编写于
8月 24, 2018
作者:
赵
赵奇可
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
resize bicubic implement
上级
b7c1990e
变更
4
隐藏空白更改
内联
并排
Showing
4 changed file
with
20 addition
and
0 deletion
+20
-0
mace/ops/ops_register.cc
mace/ops/ops_register.cc
+2
-0
mace/python/tools/converter_tool/base_converter.py
mace/python/tools/converter_tool/base_converter.py
+1
-0
mace/python/tools/converter_tool/tensorflow_converter.py
mace/python/tools/converter_tool/tensorflow_converter.py
+16
-0
repository/opencl-kernel/opencl_kernel_configure.bzl
repository/opencl-kernel/opencl_kernel_configure.bzl
+1
-0
未找到文件。
mace/ops/ops_register.cc
浏览文件 @
d804bca8
...
...
@@ -47,6 +47,7 @@ extern void Register_Proposal(OperatorRegistryBase *op_registry);
extern
void
Register_Quantize
(
OperatorRegistryBase
*
op_registry
);
extern
void
Register_ReduceMean
(
OperatorRegistryBase
*
op_registry
);
extern
void
Register_Reshape
(
OperatorRegistryBase
*
op_registry
);
extern
void
Register_ResizeBicubic
(
OperatorRegistryBase
*
op_registry
);
extern
void
Register_ResizeBilinear
(
OperatorRegistryBase
*
op_registry
);
extern
void
Register_Shape
(
OperatorRegistryBase
*
op_registry
);
extern
void
Register_Split
(
OperatorRegistryBase
*
op_registry
);
...
...
@@ -98,6 +99,7 @@ OperatorRegistry::OperatorRegistry() : OperatorRegistryBase() {
ops
::
Register_Quantize
(
this
);
ops
::
Register_ReduceMean
(
this
);
ops
::
Register_Reshape
(
this
);
ops
::
Register_ResizeBicubic
(
this
);
ops
::
Register_ResizeBilinear
(
this
);
ops
::
Register_Shape
(
this
);
ops
::
Register_Split
(
this
);
...
...
mace/python/tools/converter_tool/base_converter.py
浏览文件 @
d804bca8
...
...
@@ -100,6 +100,7 @@ MaceSupportedOps = [
'Quantize'
,
'ReduceMean'
,
'Reshape'
,
'ResizeBicubic'
,
'ResizeBilinear'
,
'Slice'
,
'Split'
,
...
...
mace/python/tools/converter_tool/tensorflow_converter.py
浏览文件 @
d804bca8
...
...
@@ -80,6 +80,7 @@ TFSupportedOps = [
'Shape'
,
'Transpose'
,
'Softmax'
,
'ResizeBicubic'
,
'ResizeBilinear'
,
'Placeholder'
,
'SpaceToBatchND'
,
...
...
@@ -178,6 +179,7 @@ class TensorflowConverter(base_converter.ConverterInterface):
TFOpType
.
Squeeze
.
name
:
self
.
convert_squeeze
,
TFOpType
.
Transpose
.
name
:
self
.
convert_transpose
,
TFOpType
.
Softmax
.
name
:
self
.
convert_softmax
,
TFOpType
.
ResizeBicubic
.
name
:
self
.
convert_resize_bicubic
,
TFOpType
.
ResizeBilinear
.
name
:
self
.
convert_resize_bilinear
,
TFOpType
.
Placeholder
.
name
:
self
.
convert_nop
,
TFOpType
.
SpaceToBatchND
.
name
:
self
.
convert_space_batch
,
...
...
@@ -511,6 +513,20 @@ class TensorflowConverter(base_converter.ConverterInterface):
op
=
self
.
convert_general_op
(
tf_op
)
op
.
type
=
MaceOp
.
Softmax
.
name
def
convert_resize_bicubic
(
self
,
tf_op
):
op
=
self
.
convert_general_op
(
tf_op
)
op
.
type
=
MaceOp
.
ResizeBicubic
.
name
del
op
.
input
[
1
:]
size_arg
=
op
.
arg
.
add
()
size_arg
.
name
=
MaceKeyword
.
mace_resize_size_str
size_value
=
tf_op
.
inputs
[
1
].
eval
().
astype
(
np
.
int32
)
size_arg
.
ints
.
extend
(
size_value
)
self
.
_skip_tensor
.
add
(
tf_op
.
inputs
[
1
].
name
)
align_corners_arg
=
op
.
arg
.
add
()
align_corners_arg
.
name
=
MaceKeyword
.
mace_align_corners_str
align_corners_arg
.
i
=
tf_op
.
get_attr
(
tf_align_corners
)
def
convert_resize_bilinear
(
self
,
tf_op
):
op
=
self
.
convert_general_op
(
tf_op
)
op
.
type
=
MaceOp
.
ResizeBilinear
.
name
...
...
repository/opencl-kernel/opencl_kernel_configure.bzl
浏览文件 @
d804bca8
...
...
@@ -41,6 +41,7 @@ def _opencl_encrypt_kernel_impl(repository_ctx):
unused_var
=
repository_ctx
.
path
(
Label
(
"//:mace/kernels/opencl/cl/pad.cl"
))
unused_var
=
repository_ctx
.
path
(
Label
(
"//:mace/kernels/opencl/cl/pooling.cl"
))
unused_var
=
repository_ctx
.
path
(
Label
(
"//:mace/kernels/opencl/cl/reduce_mean.cl"
))
unused_var
=
repository_ctx
.
path
(
Label
(
"//:mace/kernels/opencl/cl/resize_bicubic.cl"
))
unused_var
=
repository_ctx
.
path
(
Label
(
"//:mace/kernels/opencl/cl/resize_bilinear.cl"
))
unused_var
=
repository_ctx
.
path
(
Label
(
"//:mace/kernels/opencl/cl/split.cl"
))
unused_var
=
repository_ctx
.
path
(
Label
(
"//:mace/kernels/opencl/cl/softmax.cl"
))
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录