I found with the MergeComposite upgrade, it will take the model variables inside(like weights and bias) as ConstantNode in the codegen.cc rather than the former VarNode. This is more reasonable but I have some questions here:
- Where does the constant conversion happen? As if I print the mod after the partition I can still see the variables with names. And if I convert the call node into the function node, I can get the function node’s params with their name_hints. They are still VarNodes here.
- With the ConstantNode type, I cannot get the variable name anymore, how can I identify which variable it is? In the VisitExpr_ for ConstantNode here, we get the variables from the consts array, but how the order is decided?
Appreciate for any experts explaining for the
ConstantNode processing here!