-
Notifications
You must be signed in to change notification settings - Fork 27
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
How To Update Graphsurgeon Converter #15
Comments
Just add these lines directly to the create_node function. def create_node(name, op=None, trt_plugin=False, **kwargs): |
Hello @ZouYunzhe
When we update the node_manipulation.py file, where do we add these lines: We add the rest in the create_node function, however adding the above lines at the top of the document results in errors while importing graphsurgeoun to our notebook, and only adding the lines to the create_node function results in the TypeError: Cannot convert value 0 to a TensorFlow DType. It would be a huge help if you could share what your node_manipulation.py file looked like after you're done with the changes. |
how do I update graphsurgeon converter I am confused to put the code that has been provided
diff --git a/node_manipulation.py b/node_manipulation.py
index d2d012a..1ef30a0 100644
--- a/node_manipulation.py
+++ b/node_manipulation.py
@@ -30,6 +30,7 @@ def create_node(name, op=None, _do_suffix=False, **kwargs):
node = NodeDef()
node.name = name
node.op = op if op else name
for key, val in kwargs.items():
if key == "dtype":
node.attr["dtype"].type = val.as_datatype_enu
The text was updated successfully, but these errors were encountered: