Searched refs:_add_control_input (Results 1 – 13 of 13) sorted by relevance
302 switch_op._add_control_input(last_write_to_resource[input_id])306 merge_for_resource[input_id]._add_control_input(new_merge[0].op)400 op._add_control_input(o)434 merge_for_resource[input_id]._add_control_input(op)445 op._add_control_input(last_write_to_resource[None])453 op._add_control_input(collective_manager_scopes_opened[manager_id])459 op._add_control_input(collective_manager_scopes_used[manager_id])
958 z._add_control_input(x) # pylint: disable=protected-access960 z._add_control_input(x) # pylint: disable=protected-access998 y.op._add_control_input(z.op) # pylint: disable=protected-access999 y.op._add_control_input(x.op) # pylint: disable=protected-access1000 x.op._add_control_input(y.op) # pylint: disable=protected-access1096 while_op._add_control_input(constant_op.constant(0).op)
2321 def _add_control_input(self, op): member in Operation
234 self._forward_index.op._add_control_input(self._forward_sync)258 self._grad_index.op._add_control_input(self._grad_sync)363 self.forward_index.op._add_control_input(push.op)382 self.forward_sync._add_control_input(push.op)385 push.op._add_control_input(add_op)434 self.grad_sync._add_control_input(pop.op)
995 op._add_control_input(self._pivot.op)1024 op._add_control_input(self._pivot.op)1776 op._add_control_input(self.GetControlPivot().op) # pylint: disable=protected-access1783 op._add_control_input(self.GetControlPivot().op)1838 op._add_control_input(self.GetControlPivot().op)1865 n.op._add_control_input(outer_add_op) # pylint: disable=protected-access1949 outer_grad_state.grad_sync._add_control_input(final_zero.op)2188 var.op._add_control_input(control_pivot.op)
374 lock_op._add_control_input(all_args)
259 outputs[0].op._add_control_input( # pylint: disable=protected-access
329 identity.op._add_control_input( # pylint: disable=protected-access571 outputs[0].op._add_control_input( # pylint: disable=protected-access
358 mutation.copied_op._add_control_input(op_map[mutation.old_graph_op])
3500 outputs[0].op._add_control_input(printer)
237 op._add_control_input(self._pivot)
612 op._add_control_input(self.GetControlPivot())
1877 consumer_op._add_control_input(trace_op)