Searched refs:_set_control_flow_context (Results 1 – 8 of 8) sorted by relevance
53 graph._set_control_flow_context(ctx.outer_context) # pylint: disable=protected-access55 graph._set_control_flow_context(saved_context) # pylint: disable=protected-access
229 self._forward_sync._set_control_flow_context(self._forward_context)253 self._grad_sync._set_control_flow_context(self._grad_context)371 push.op._set_control_flow_context(value_ctxt)695 val.op._set_control_flow_context(op_ctxt)696 zeros_shape.op._set_control_flow_context(op_ctxt)
244 ops.get_default_graph()._set_control_flow_context(control_flow_context)
634 g.as_graph_element(op)._set_control_flow_context(self)689 graph._set_control_flow_context(self)695 graph._set_control_flow_context(last_context)804 pivot.op._set_control_flow_context(self) # pylint: disable=protected-access914 result.op._set_control_flow_context(self)1720 op._set_control_flow_context(op_input_ctxt)2326 x.op._set_control_flow_context(self)
48 graph._set_control_flow_context(
1933 graph._set_control_flow_context(op_control_flow_context)1986 graph._set_control_flow_context(self._outmost_context)2000 graph._set_control_flow_context(self._outmost_context)
396 graph._set_control_flow_context(self.outer_context)401 graph._set_control_flow_context(saved_context)
2236 def _set_control_flow_context(self, ctx): member in Operation3469 def _set_control_flow_context(self, ctx): member in Graph4906 self._graph._set_control_flow_context(None)4913 self._graph._set_control_flow_context(self._old_control_flow_context)