|
|
@@ -499,6 +499,10 @@ class DavinciModel { |
|
|
|
return exception_dumper_.DumpExceptionInfo(exception_infos); |
|
|
|
} |
|
|
|
|
|
|
|
void SetKnownShapeGlobalStep(void *global_step) { |
|
|
|
known_shape_global_step_ = global_step; |
|
|
|
} |
|
|
|
|
|
|
|
void DumperShrink() { |
|
|
|
data_dumper_.DumpShrink(); |
|
|
|
} |
|
|
@@ -1105,6 +1109,9 @@ class DavinciModel { |
|
|
|
vector<InputOutputDescInfo> output_descs_; |
|
|
|
vector<uint32_t> output_formats_; |
|
|
|
|
|
|
|
// known shape node for dump |
|
|
|
void *known_shape_global_step_; |
|
|
|
|
|
|
|
// op name to attrs mapping |
|
|
|
std::map<std::string, std::map<std::string, std::vector<std::string>>> op_name_to_attrs_; |
|
|
|
}; |
|
|
|