diff options
| -rw-r--r-- | tensorflow/python/ops/control_flow_ops.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/python/ops/control_flow_ops.py b/tensorflow/python/ops/control_flow_ops.py index 4b9b34b49d..d8a538c4e3 100644 --- a/tensorflow/python/ops/control_flow_ops.py +++ b/tensorflow/python/ops/control_flow_ops.py @@ -1496,7 +1496,8 @@ class ControlFlowContext(object): def AddInnerOp(self, op): """Notifies a scope about an operator added to an inner scope.""" - pass + if self._outer_context: + self._outer_context.AddInnerOp(op) def GetControlPivot(self): """Returns the pivot node for this context, or None.""" |