Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
magicwindyyd
mindspore
提交
de8db1ab
M
mindspore
项目概览
magicwindyyd
/
mindspore
与 Fork 源项目一致
Fork自
MindSpore / mindspore
通知
1
Star
1
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
M
mindspore
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
de8db1ab
编写于
8月 08, 2020
作者:
S
sunsuodong
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
support leaky relu
上级
8e5d4445
变更
4
隐藏空白更改
内联
并排
Showing
4 changed file
with
8 addition
and
10 deletion
+8
-10
mindspore/lite/schema/ops.fbs
mindspore/lite/schema/ops.fbs
+1
-0
mindspore/lite/src/populate_parameter.cc
mindspore/lite/src/populate_parameter.cc
+1
-0
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/activation.h
...spore/lite/src/runtime/kernel/arm/nnacl/fp32/activation.h
+1
-1
mindspore/lite/tools/converter/parser/caffe/caffe_relu_parser.cc
...re/lite/tools/converter/parser/caffe/caffe_relu_parser.cc
+5
-9
未找到文件。
mindspore/lite/schema/ops.fbs
浏览文件 @
de8db1ab
...
...
@@ -142,6 +142,7 @@ table SoftMax {
table Activation {
type: ActivationType = 0;
alpha: float = 0.2;
}
table ActivationGrad {
type: ActivationGradType = 0;
...
...
mindspore/lite/src/populate_parameter.cc
浏览文件 @
de8db1ab
...
...
@@ -487,6 +487,7 @@ OpParameter *PopulateActivationParameter(const lite::Primitive *primitive) {
}
auto
activation
=
primitive
->
Value
()
->
value_as_Activation
();
act_param
->
type_
=
static_cast
<
int
>
(
activation
->
type
());
act_param
->
alpha_
=
activation
->
alpha
();
return
reinterpret_cast
<
OpParameter
*>
(
act_param
);
}
...
...
mindspore/lite/src/runtime/kernel/arm/nnacl/fp32/activation.h
浏览文件 @
de8db1ab
...
...
@@ -24,7 +24,7 @@
struct
ActivationParameter
{
OpParameter
op_parameter_
;
int
type_
;
float
alpha_
{
0
.
01
};
float
alpha_
{
0
.
2
};
};
inline
int
Relu
(
const
float
*
src
,
int
length
,
float
*
dst
)
{
...
...
mindspore/lite/tools/converter/parser/caffe/caffe_relu_parser.cc
浏览文件 @
de8db1ab
...
...
@@ -25,22 +25,18 @@ STATUS CaffeReluParser::Parse(const caffe::LayerParameter &proto,
std
::
vector
<
schema
::
TensorT
*>
*
weightVec
)
{
std
::
unique_ptr
<
schema
::
ActivationT
>
attr
(
new
schema
::
ActivationT
());
attr
->
type
=
schema
::
ActivationType_RELU
;
op
->
primitive
=
std
::
make_unique
<
schema
::
PrimitiveT
>
();
op
->
primitive
->
value
.
value
=
attr
.
release
();
op
->
primitive
->
value
.
type
=
schema
::
PrimitiveType_Activation
;
// relu: negative_slope = 0, no parameter;
// leakyrelu: negative_slope != 0;
if
(
proto
.
has_relu_param
()
&&
proto
.
relu_param
().
has_negative_slope
())
{
float
negative_slope
=
proto
.
relu_param
().
negative_slope
();
if
(
0
!=
negative_slope
)
{
std
::
unique_ptr
<
schema
::
LeakyReLUT
>
attrLeakyReLu
(
new
schema
::
LeakyReLUT
());
attrLeakyReLu
->
negativeSlope
=
negative_slope
;
op
->
primitive
=
std
::
make_unique
<
schema
::
PrimitiveT
>
();
op
->
primitive
->
value
.
type
=
schema
::
PrimitiveType_LeakyReLU
;
op
->
primitive
->
value
.
value
=
attrLeakyReLu
.
release
();
attr
->
type
=
schema
::
ActivationType_LEAKY_RELU
;
attr
->
alpha
=
negative_slope
;
}
}
op
->
primitive
=
std
::
make_unique
<
schema
::
PrimitiveT
>
();
op
->
primitive
->
value
.
value
=
attr
.
release
();
op
->
primitive
->
value
.
type
=
schema
::
PrimitiveType_Activation
;
return
RET_OK
;
}
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录