12 lines
452 B
Groff
12 lines
452 B
Groff
|
diff -Naur a/node_manipulation.py b/node_manipulation.py
|
||
|
|
--- a/node_manipulation.py 2019-10-24 13:17:10.203943256 +0800
|
||
|
|
+++ b/node_manipulation.py 2019-10-24 13:19:08.851943211 +0800
|
||
|
|
@@ -39,6 +39,7 @@
|
||
|
|
'''
|
||
|
|
node.name = name or node.name
|
||
|
|
node.op = op or node.op or node.name
|
||
|
|
+ node.attr["dtype"].type = 1
|
||
|
|
for key, val in kwargs.items():
|
||
|
|
if isinstance(val, tf.DType):
|
||
|
|
node.attr[key].type = val.as_datatype_enum
|