wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-01 18:04:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

Updated tfutils to use new op naming (#284)

Browse Source 
* Updated tfutils to use new op naming

* Reverted tensorflow 12.0.0
This commit is contained in:
Wapaul1
2017-02-15 17:47:53 -08:00
committed by Robert Nishihara
parent 12a68e84d2
commit 883f945db4
2 changed files with 5 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/experimental/tfutils.py
+1 -1
View File
@@ -58,7 +58,7 @@ class TensorFlowVariables(object):
if control not in explored_inputs:
queue.append(control)
explored_inputs.add(control)
if tf_obj.node_def.op == "Variable":
if "Variable" in tf_obj.node_def.op:
variable_names.append(tf_obj.node_def.name)
self.variables = OrderedDict()
for v in [v for v in tf.global_variables() if v.op.node_def.name in variable_names]: