mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

gaojingqingli

This commit is contained in:
Henry 2022-08-05 10:37:41 +08:00
parent 6d87fa6288
commit 86eb7dbec1
3 changed files with 4 additions and 4 deletions
mindspore/ccsrc/transform
express_ir
onnx_exporter.cc
graph_ir
op_adapter.ccop_adapter.h

4
mindspore/ccsrc/transform/express_ir/onnx_exporter.cc
View File

@ -1114,7 +1114,7 @@ class OnnxExporter {
void SetValueInfoType(const AnfNodePtr &node, onnx::ValueInfoProto *value_proto, int64_t output_index = -1);
void MatchAndMark(const FuncGraphPtr &func_graph, const std::vector<AnfNodePtr> &nodes,
mindspore::HashMap<AnfNodePtr, OpMergedInfo> *op_merged_infos_ptr);
mindspore::HashMap<AnfNodePtr, OpMergedInfo> *op_merged_infos_ptr) const;
void MatchAndMarkCNode(const FuncGraphPtr &func_graph, const CNodePtr &cnode,
mindspore::HashMap<AnfNodePtr, OpMergedInfo> *op_merged_infos_ptr);
void ExportNodes(const FuncGraphPtr &func_graph, std::map<AnfNodePtr, std::string> *node_map_ptr,
@ -1389,7 +1389,7 @@ void OnnxExporter::SetValueInfoType(const AnfNodePtr &node, onnx::ValueInfoProto
}
void OnnxExporter::MatchAndMark(const FuncGraphPtr &func_graph, const std::vector<AnfNodePtr> &nodes,
mindspore::HashMap<AnfNodePtr, OpMergedInfo> *op_merged_infos_ptr) {
mindspore::HashMap<AnfNodePtr, OpMergedInfo> *op_merged_infos_ptr) const {
auto &op_merged_infos = *op_merged_infos_ptr;
for (auto &node : nodes) {

2
mindspore/ccsrc/transform/graph_ir/op_adapter.cc
View File

@ -450,7 +450,7 @@ void OpAdapterImpl::UpdateNormalOpInputDesc(const OperatorPtr &op, const AnfNode
}
void OpAdapterImpl::UpdateCustomOpInputDesc(const CusOperatorPtr &op, const AnfNodePtr &node,
const std::string format) {
const std::string format) const {
if (op == nullptr) {
MS_LOG(ERROR) << "op is nullptr";
return;

2
mindspore/ccsrc/transform/graph_ir/op_adapter.h
View File

@ -89,7 +89,7 @@ class OpAdapterImpl {
const std::string &format);
std::shared_ptr<GeTensorDesc> CreateNodeDesc(const AnfNodePtr &node, const std::string &format);
void UpdateNormalOpInputDesc(const OperatorPtr &op, const AnfNodePtr &node, const std::string format);
void UpdateCustomOpInputDesc(const CusOperatorPtr &op, const AnfNodePtr &node, const std::string format);
void UpdateCustomOpInputDesc(const CusOperatorPtr &op, const AnfNodePtr &node, const std::string format) const;
void updateInputDesc(const OperatorPtr &op, const AnfNodePtr &node);
void updateOutputDesc(const OperatorPtr &op, const abstract::BaseShapePtr &shp, const TypePtr &type,
const AnfNodePtr &node);