-
Notifications
You must be signed in to change notification settings - Fork 24
Move rename_tensor to function_hook #136
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Conversation
Codecov Report
@@ Coverage Diff @@
## master #136 +/- ##
==========================================
+ Coverage 83.14% 83.25% +0.11%
==========================================
Files 22 22
Lines 1127 1135 +8
==========================================
+ Hits 937 945 +8
Misses 190 190
Continue to review full report at Codecov.
|
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Great, thanks!
self.graph.extend(nodes) | ||
self.converted_nodes[temp_node_name] = nodes | ||
|
||
def deleted(self, function=None): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Could you add a brief comment to explain why we need to use deleted
?
onnx_chainer/export.py
Outdated
When renaming an output name, another node can reference the same value | ||
as input, so the input name must be renamed at once. So this renaming | ||
process should be run after all functions are converted and this | ||
`deleted` function is called when function hook is done, means |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
How about "which means all functions are converted"?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
fixed, thx
onnx_chainer/export.py
Outdated
`deleted` function is called when function hook is done, means | ||
completed functions converting. | ||
|
||
If input/output name is given by externally, these given names take |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Maybe: "names are given externally"
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
fixed, thx
Move
rename_tensor
function toFunctionHook.deleted
function, to be more customizable.FunctionNode
name, not ONNX operator name. This change breaks backward compatibilityFunctionNode
name as node name