Browse Source

!8129 add uint for SetValueToProto

Merge pull request !8129 from Simson/bugfix
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
06838ded82
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      mindspore/ccsrc/debug/dump_proto.cc

+ 3
- 0
mindspore/ccsrc/debug/dump_proto.cc View File

@@ -185,6 +185,9 @@ void ProtoExporter::SetValueToProto(const ValuePtr &val, irpb::ValueProto *value
} else if (val->isa<Int>()) {
value_proto->set_dtype(irpb::DT_TYPE);
value_proto->mutable_type_val()->set_data_type(irpb::DT_BASE_INT);
} else if (val->isa<UInt>()) {
value_proto->set_dtype(irpb::DT_TYPE);
value_proto->mutable_type_val()->set_data_type(irpb::DT_BASE_UINT);
} else if (val->isa<Float>()) {
value_proto->set_dtype(irpb::DT_TYPE);
value_proto->mutable_type_val()->set_data_type(irpb::DT_BASE_FLOAT);


Loading…
Cancel
Save