Searched refs:_set_control_flow_context (Results 1 – 5 of 5) sorted by relevance
772 self._forward_sync._set_control_flow_context(self._forward_context)796 self._grad_sync._set_control_flow_context(self._grad_context)914 push.op._set_control_flow_context(value_ctxt)1239 val.op._set_control_flow_context(op_ctxt)1240 zeros_shape.op._set_control_flow_context(op_ctxt)1412 g.as_graph_element(op)._set_control_flow_context(self)1467 graph._set_control_flow_context(self)1473 graph._set_control_flow_context(last_context)1589 pivot.op._set_control_flow_context(self) # pylint: disable=protected-access1699 result.op._set_control_flow_context(self)[all …]
194 graph._set_control_flow_context(self.outer_context)197 graph._set_control_flow_context(saved_context)
1489 graph._set_control_flow_context(op_control_flow_context)1519 graph._set_control_flow_context(current_control_flow_context)
3380 self._g._set_control_flow_context(self)3384 self._g._set_control_flow_context(self._old) # pylint: disable=protected-access
2079 def _set_control_flow_context(self, ctx): member in Operation3235 def _set_control_flow_context(self, ctx): member in Graph4529 self._graph._set_control_flow_context(None)4536 self._graph._set_control_flow_context(self._old_control_flow_context)