|
|
|
@@ -1135,6 +1135,9 @@ void Debugger::LoadSingleAnfnode(const AnfNodePtr &anf_node, const size_t output |
|
|
|
auto addr = AnfAlgo::GetOutputAddr(anf_node, output_index); |
|
|
|
MS_EXCEPTION_IF_NULL(addr); |
|
|
|
auto type = AnfAlgo::GetOutputInferDataType(anf_node, output_index); |
|
|
|
if (type == kObjectTypeUMonad) { |
|
|
|
return; |
|
|
|
} |
|
|
|
auto format = kOpFormat_DEFAULT; |
|
|
|
string tensor_name = node_name + ':' + "0"; |
|
|
|
ShapeVector int_shapes; |
|
|
|
@@ -1197,6 +1200,9 @@ void Debugger::LoadGraphOutputs() { |
|
|
|
auto addr = AnfAlgo::GetOutputAddr(node, j); |
|
|
|
MS_EXCEPTION_IF_NULL(addr); |
|
|
|
auto type = AnfAlgo::GetOutputInferDataType(node, j); |
|
|
|
if (type == kObjectTypeUMonad) { |
|
|
|
continue; |
|
|
|
} |
|
|
|
auto format = kOpFormat_DEFAULT; |
|
|
|
string tensor_name = kernel_name + ':' + std::to_string(j); |
|
|
|
ShapeVector int_shapes; |
|
|
|
|