experiments
/
STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Replace list concatenation with generators for auto control deps

This commit is contained in:
Lukas Geiger 2020-05-26 22:43:28 +01:00
parent e7cc47384f
commit 4c90f20c0d
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/python/framework/auto_control_deps.py
View File

@ -535,8 +535,10 @@ def _get_resource_inputs(op):
# Note: A resource handle that is not written to is treated as read-only. We # Note: A resource handle that is not written to is treated as read-only. We
# don't have a special way of denoting an unused resource. # don't have a special way of denoting an unused resource.
return ([(t, ResourceType.READ_ONLY) for t in reads] + for t in reads:
[(t, ResourceType.READ_WRITE) for t in writes]) yield (t, ResourceType.READ_ONLY)
for t in writes:
yield (t, ResourceType.READ_WRITE)
def automatic_control_dependencies(f): def automatic_control_dependencies(f):