|
|
|
@ -40,9 +40,7 @@ void CheckIfValidType(const TypePtr &type, debugger::TypeProto *type_proto) {
|
|
|
|
|
type->isa<UMonadType>() || type->isa<IOMonadType>())) {
|
|
|
|
|
MS_LOG(EXCEPTION) << "Unknown type: " << type->type_name();
|
|
|
|
|
}
|
|
|
|
|
if (type == nullptr) {
|
|
|
|
|
type_proto->set_data_type(debugger::DT_UNDEFINED);
|
|
|
|
|
} else if (type->isa<Number>()) {
|
|
|
|
|
if (type->isa<Number>()) {
|
|
|
|
|
type_proto->set_data_type(GetDebuggerNumberDataType(type));
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
@ -52,9 +50,11 @@ void DebuggerProtoExporter::SetNodeOutputType(const TypePtr &type, const BaseSha
|
|
|
|
|
if (type_proto == nullptr) {
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
if (type != nullptr) {
|
|
|
|
|
CheckIfValidType(type, type_proto);
|
|
|
|
|
if (type == nullptr) {
|
|
|
|
|
type_proto->set_data_type(debugger::DT_UNDEFINED);
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
CheckIfValidType(type, type_proto);
|
|
|
|
|
if (type->isa<TensorType>()) {
|
|
|
|
|
TypePtr elem_type = dyn_cast<TensorType>(type)->element();
|
|
|
|
|
type_proto->mutable_tensor_type()->set_elem_type(GetDebuggerNumberDataType(elem_type));
|
|
|
|
|