|
|
@@ -111,12 +111,12 @@ KernelWithIndex AnfRuntimeAlgorithm::VisitKernelWithReturnType(const AnfNodePtr |
|
|
MS_EXCEPTION_IF_NULL(value_node); |
|
|
MS_EXCEPTION_IF_NULL(value_node); |
|
|
int item_idx = GetValue<int>(value_node->value()); |
|
|
int item_idx = GetValue<int>(value_node->value()); |
|
|
return VisitKernelWithReturnType(cnode->input(kRealInputNodeIndexInTupleGetItem), IntToSize(item_idx), |
|
|
return VisitKernelWithReturnType(cnode->input(kRealInputNodeIndexInTupleGetItem), IntToSize(item_idx), |
|
|
visit_nop_node); |
|
|
|
|
|
|
|
|
visit_nop_node, return_types); |
|
|
} else if (IsPrimitive(input0, prim::kPrimDepend) || IsPrimitive(input0, prim::kPrimControlDepend)) { |
|
|
} else if (IsPrimitive(input0, prim::kPrimDepend) || IsPrimitive(input0, prim::kPrimControlDepend)) { |
|
|
return VisitKernelWithReturnType(cnode->input(kRealInputIndexInDepend), 0, visit_nop_node); |
|
|
|
|
|
|
|
|
return VisitKernelWithReturnType(cnode->input(kRealInputIndexInDepend), 0, visit_nop_node, return_types); |
|
|
} else if (opt::IsNopNode(cnode) && visit_nop_node) { |
|
|
} else if (opt::IsNopNode(cnode) && visit_nop_node) { |
|
|
if (cnode->inputs().size() == 2) { |
|
|
if (cnode->inputs().size() == 2) { |
|
|
return VisitKernelWithReturnType(cnode->input(1), 0, visit_nop_node); |
|
|
|
|
|
|
|
|
return VisitKernelWithReturnType(cnode->input(1), 0, visit_nop_node, return_types); |
|
|
} else { |
|
|
} else { |
|
|
MS_LOG(EXCEPTION) << cnode->DebugString() << "Invalid nop node"; |
|
|
MS_LOG(EXCEPTION) << cnode->DebugString() << "Invalid nop node"; |
|
|
} |
|
|
} |
|
|
|