Searched refs:_add_control_input (Results 1 – 8 of 8) sorted by relevance
221 …switch_op._add_control_input(last_op_using_resource_tensor[inp]) # pylint: disable=protected-acce…225 … merge_for_resource[inp]._add_control_input(new_merge[0].op) # pylint: disable=protected-access298 op._add_control_input(o) # pylint: disable=protected-access325 merge_for_resource[inp]._add_control_input(op) # pylint: disable=protected-access330 … op._add_control_input(last_op_using_resource_tensor[None]) # pylint: disable=protected-access
481 z._add_control_input(x) # pylint: disable=protected-access483 z._add_control_input(x) # pylint: disable=protected-access521 y.op._add_control_input(z.op) # pylint: disable=protected-access522 y.op._add_control_input(x.op) # pylint: disable=protected-access523 x.op._add_control_input(y.op) # pylint: disable=protected-access619 while_op._add_control_input(constant_op.constant(0).op)
2288 def _add_control_input(self, op): member in Operation
773 self._forward_index.op._add_control_input(self._forward_sync)797 self._grad_index.op._add_control_input(self._grad_sync)902 self.forward_index.op._add_control_input(push.op)921 self.forward_sync._add_control_input(push.op)924 push.op._add_control_input(add_op)972 self.grad_sync._add_control_input(pop.op)1727 op._add_control_input(self._pivot.op)1756 op._add_control_input(self._pivot.op)2483 op._add_control_input(self.GetControlPivot().op)2538 op._add_control_input(self.GetControlPivot().op)[all …]
327 lock_op._add_control_input(all_args)
208 op._add_control_input(self._pivot)
395 op._add_control_input(self.GetControlPivot())
1515 consumer_op._add_control_input(trace_op)