Searched refs:_add_control_input (Results 1 – 13 of 13) sorted by relevance
337 switch_op._add_control_input(last_write_to_resource[input_id])341 merge_for_resource[input_id]._add_control_input(new_merge[0].op)453 op._add_control_input(o)487 merge_for_resource[input_id]._add_control_input(op)524 op._add_control_input(last_write_to_resource[None])532 op._add_control_input(collective_manager_scopes_opened[manager_id])538 op._add_control_input(collective_manager_scopes_used[manager_id])
982 z._add_control_input(x) # pylint: disable=protected-access984 z._add_control_input(x) # pylint: disable=protected-access1022 y.op._add_control_input(z.op) # pylint: disable=protected-access1023 y.op._add_control_input(x.op) # pylint: disable=protected-access1024 x.op._add_control_input(y.op) # pylint: disable=protected-access1120 while_op._add_control_input(constant_op.constant(0).op)
2459 def _add_control_input(self, op): member in Operation
230 self._forward_index.op._add_control_input(self._forward_sync)254 self._grad_index.op._add_control_input(self._grad_sync)359 self.forward_index.op._add_control_input(push.op)378 self.forward_sync._add_control_input(push.op)381 push.op._add_control_input(add_op)430 self.grad_sync._add_control_input(pop.op)
942 op._add_control_input(self._pivot.op)971 op._add_control_input(self._pivot.op)1734 op._add_control_input(self.GetControlPivot().op) # pylint: disable=protected-access1741 op._add_control_input(self.GetControlPivot().op)1796 op._add_control_input(self.GetControlPivot().op)1823 n.op._add_control_input(outer_add_op) # pylint: disable=protected-access1907 outer_grad_state.grad_sync._add_control_input(final_zero.op)2161 var.op._add_control_input(control_pivot.op)
372 lock_op._add_control_input(all_args)
323 identity.op._add_control_input( # pylint: disable=protected-access562 outputs[0].op._add_control_input( # pylint: disable=protected-access
255 outputs[0].op._add_control_input( # pylint: disable=protected-access
361 mutation.copied_op._add_control_input(op_map[mutation.old_graph_op])
3682 outputs[0].op._add_control_input(printer)
232 op._add_control_input(self._pivot)
635 op._add_control_input(self.GetControlPivot())
1996 consumer_op._add_control_input(trace_op)