Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
magicwindyyd
mindspore
提交
25a878d6
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看板
提交
25a878d6
编写于
9月 05, 2020
作者:
M
mindspore-ci-bot
提交者:
Gitee
9月 05, 2020
浏览文件
操作
浏览文件
下载
差异文件
!5719 [MSLITE]mindspore export inferface finetune lite
Merge pull request !5719 from zhengjun10/master
上级
f91d80e0
22dc3405
变更
3
展开全部
隐藏空白更改
内联
并排
Showing
3 changed file
with
264 addition
and
174 deletion
+264
-174
mindspore/lite/test/models_mindspore.cfg
mindspore/lite/test/models_mindspore.cfg
+2
-2
mindspore/lite/tools/anf_importer/import_from_protobuf.cc
mindspore/lite/tools/anf_importer/import_from_protobuf.cc
+255
-162
mindspore/lite/tools/anf_importer/import_from_protobuf.h
mindspore/lite/tools/anf_importer/import_from_protobuf.h
+7
-10
未找到文件。
mindspore/lite/test/models_mindspore.cfg
浏览文件 @
25a878d6
ssd.pb
mobilenet_v2.pb
#
ssd.pb
#
mobilenet_v2.pb
mindspore/lite/tools/anf_importer/import_from_protobuf.cc
浏览文件 @
25a878d6
此差异已折叠。
点击以展开。
mindspore/lite/tools/anf_importer/import_from_protobuf.h
浏览文件 @
25a878d6
...
...
@@ -42,9 +42,9 @@ class AnfImporterFromProtobuf : public AnfImporter {
int
Import
(
const
schema
::
QuantType
&
quantType
=
schema
::
QuantType_QUANT_NONE
)
override
;
private:
int
ConverterConstTensor
()
override
{
return
RET_ERROR
;
};
int
ConverterCNode
()
override
{
return
RET_ERROR
;
};
int
AddReturnCNode
()
override
{
return
RET_ERROR
;
};
int
ConverterConstTensor
()
override
{
return
RET_ERROR
;
};
int
ConverterCNode
()
override
{
return
RET_ERROR
;
};
int
AddReturnCNode
()
override
{
return
RET_ERROR
;
};
bool
ParseModelConfigureInfo
(
const
onnx
::
ModelProto
&
model_proto
);
bool
BuildFuncGraph
(
const
FuncGraphPtr
&
outputFuncGraph
,
const
onnx
::
GraphProto
&
importProto
,
const
schema
::
QuantType
&
quantType
);
...
...
@@ -59,18 +59,15 @@ class AnfImporterFromProtobuf : public AnfImporter {
bool
GetAttrValueForCNode
(
const
PrimitivePtr
&
prim
,
const
onnx
::
AttributeProto
&
attr_proto
);
bool
ObtainCNodeAttrInTypeForm
(
const
PrimitivePtr
&
prim
,
const
std
::
string
&
attr_name
,
const
onnx
::
TensorProto
&
attr_tensor
);
bool
ObtainCNodeAttrInScalarForm
(
const
PrimitivePtr
&
prim
,
const
std
::
string
&
attr_name
,
const
onnx
::
TensorProto
&
attr_tensor
);
ValuePtr
ObtainCNodeAttrInScalarForm
(
const
onnx
::
TensorProto
&
attr_tensor
);
bool
ObtainCNodeAttrInTensorForm
(
const
PrimitivePtr
&
prim
,
const
std
::
string
&
attr_name
,
const
onnx
::
TensorProto
&
attr_tensor
);
bool
BuildValueNodeForFuncGraph
(
const
onnx
::
NodeProto
&
node_proto
);
bool
ObtainValueNodeInTensorForm
(
const
string
&
value_node_name
,
const
onnx
::
TensorProto
&
attr_tensor
);
bool
ObtainValueNodeInScalarForm
(
const
string
&
value_node_name
,
const
onnx
::
TensorProto
&
attr_tensor
);
bool
GetAttrValueForValueNode
(
const
string
&
ref_attr_name
,
const
std
::
string
&
value_node_name
,
const
onnx
::
TensorProto
&
attr_tensor
);
bool
GetAttrValueForValueNode
(
const
std
::
string
&
value_node_name
,
const
onnx
::
AttributeProto
&
attr_proto
);
bool
ObtainValueNodeInTypeForm
(
const
string
&
value_node_name
,
const
onnx
::
TensorProto
&
attr_tensor
);
abstract
::
AbstractTensorPtr
GetAbstractForCNode
(
const
onnx
::
AttributeProto
&
attr_proto
);
std
::
unordered_map
<
std
::
string
,
abstract
::
AbstractTensorPtr
>
GetAbstractForCNode
(
const
onnx
::
AttributeProto
&
attr_proto
);
private:
std
::
string
producer_name_
;
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录