Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
magicwindyyd
mindspore
提交
84f2c57e
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看板
提交
84f2c57e
编写于
8月 10, 2020
作者:
M
mindspore-ci-bot
提交者:
Gitee
8月 10, 2020
浏览文件
操作
浏览文件
下载
差异文件
!4184 fix CopyPrimitive and leaky_relu_parser
Merge pull request !4184 from sunsuodong/fix_resize
上级
3b905d9c
5a7863de
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
10 addition
and
3 deletion
+10
-3
mindspore/lite/src/model_impl.cc
mindspore/lite/src/model_impl.cc
+6
-0
mindspore/lite/tools/converter/parser/tflite/tflite_leaky_relu_parser.cc
...tools/converter/parser/tflite/tflite_leaky_relu_parser.cc
+4
-3
未找到文件。
mindspore/lite/src/model_impl.cc
浏览文件 @
84f2c57e
...
...
@@ -166,6 +166,8 @@ lite::Primitive *ModelImpl::CopyPrimitive(const schema::Primitive *srcPrim) {
return
new
lite
::
Exp
(
const_cast
<
schema
::
Primitive
*>
(
srcPrim
));
case
schema
::
PrimitiveType_Gather
:
return
new
lite
::
Gather
(
const_cast
<
schema
::
Primitive
*>
(
srcPrim
));
case
schema
::
PrimitiveType_GatherNd
:
return
new
lite
::
GatherNd
(
const_cast
<
schema
::
Primitive
*>
(
srcPrim
));
case
schema
::
PrimitiveType_LocalResponseNormalization
:
return
new
lite
::
LocalResponseNormalization
(
const_cast
<
schema
::
Primitive
*>
(
srcPrim
));
case
schema
::
PrimitiveType_Maximum
:
...
...
@@ -180,6 +182,8 @@ lite::Primitive *ModelImpl::CopyPrimitive(const schema::Primitive *srcPrim) {
return
new
lite
::
Prelu
(
const_cast
<
schema
::
Primitive
*>
(
srcPrim
));
case
schema
::
PrimitiveType_Round
:
return
new
lite
::
Round
(
const_cast
<
schema
::
Primitive
*>
(
srcPrim
));
case
schema
::
PrimitiveType_Reverse
:
return
new
lite
::
Reverse
(
const_cast
<
schema
::
Primitive
*>
(
srcPrim
));
case
schema
::
PrimitiveType_ReverseSequence
:
return
new
lite
::
ReverseSequence
(
const_cast
<
schema
::
Primitive
*>
(
srcPrim
));
case
schema
::
PrimitiveType_LogicalAnd
:
...
...
@@ -212,6 +216,8 @@ lite::Primitive *ModelImpl::CopyPrimitive(const schema::Primitive *srcPrim) {
return
new
lite
::
Split
(
const_cast
<
schema
::
Primitive
*>
(
srcPrim
));
case
schema
::
PrimitiveType_OneHot
:
return
new
lite
::
OneHot
(
const_cast
<
schema
::
Primitive
*>
(
srcPrim
));
case
schema
::
PrimitiveType_Resize
:
return
new
lite
::
Resize
(
const_cast
<
schema
::
Primitive
*>
(
srcPrim
));
case
schema
::
PrimitiveType_MatMul
:
return
new
lite
::
MatMul
(
const_cast
<
schema
::
Primitive
*>
(
srcPrim
));
case
schema
::
PrimitiveType_QuantDTypeCast
:
...
...
mindspore/lite/tools/converter/parser/tflite/tflite_leaky_relu_parser.cc
浏览文件 @
84f2c57e
...
...
@@ -26,18 +26,19 @@ STATUS TfliteLeakyReluParser::Parse(const std::unique_ptr<tflite::OperatorT> &tf
const
std
::
vector
<
std
::
unique_ptr
<
tflite
::
OperatorCodeT
>>
&
tfliteOpSet
,
schema
::
CNodeT
*
op
,
TensorCache
*
tensor_cache
,
bool
quantizedModel
)
{
MS_LOG
(
DEBUG
)
<<
"parse TfliteLeakyReluParser"
;
std
::
unique_ptr
<
schema
::
LeakyReLUT
>
attr
(
new
schema
::
LeakyReLU
T
());
std
::
unique_ptr
<
schema
::
ActivationT
>
attr
(
new
schema
::
Activation
T
());
const
auto
&
tflite_attr
=
tfliteOp
->
builtin_options
.
AsLeakyReluOptions
();
if
(
tflite_attr
==
nullptr
)
{
MS_LOG
(
ERROR
)
<<
"get op: "
<<
op
->
name
.
c_str
()
<<
" attr failed"
;
return
RET_NULL_PTR
;
}
attr
->
negativeSlope
=
tflite_attr
->
alpha
;
attr
->
type
=
schema
::
ActivationType_LEAKY_RELU
;
attr
->
alpha
=
tflite_attr
->
alpha
;
if
(
op
!=
nullptr
)
{
op
->
primitive
=
std
::
make_unique
<
schema
::
PrimitiveT
>
();
op
->
primitive
->
value
.
type
=
schema
::
PrimitiveType_
LeakyReLU
;
op
->
primitive
->
value
.
type
=
schema
::
PrimitiveType_
Activation
;
op
->
primitive
->
value
.
value
=
attr
.
release
();
}
return
RET_OK
;
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录