Home
last modified time | relevance | path

Searched refs:_add_control_input (Results 1 – 13 of 13) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dauto_control_deps.py337 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])
H A Dops_test.py982 z._add_control_input(x) # pylint: disable=protected-access
984 z._add_control_input(x) # pylint: disable=protected-access
1022 y.op._add_control_input(z.op) # pylint: disable=protected-access
1023 y.op._add_control_input(x.op) # pylint: disable=protected-access
1024 x.op._add_control_input(y.op) # pylint: disable=protected-access
1120 while_op._add_control_input(constant_op.constant(0).op)
H A Dops.py2459 def _add_control_input(self, op): member in Operation
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dcontrol_flow_state.py230 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)
H A Dcontrol_flow_ops.py942 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-access
1741 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-access
1907 outer_grad_state.grad_sync._add_control_input(final_zero.op)
2161 var.op._add_control_input(control_pivot.op)
H A Dcritical_section_ops.py372 lock_op._add_control_input(all_args)
/aosp_15_r20/external/tensorflow/tensorflow/python/debug/lib/
H A Ddumping_callback.py323 identity.op._add_control_input( # pylint: disable=protected-access
562 outputs[0].op._add_control_input( # pylint: disable=protected-access
H A Dcheck_numerics_callback.py255 outputs[0].op._add_control_input( # pylint: disable=protected-access
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dlift_to_graph.py361 mutation.copied_op._add_control_input(op_map[mutation.old_graph_op])
H A Dfunction_test.py3682 outputs[0].op._add_control_input(printer)
/aosp_15_r20/external/tensorflow/tensorflow/python/compiler/xla/
H A Dxla.py232 op._add_control_input(self._pivot)
/aosp_15_r20/external/tensorflow/tensorflow/python/tpu/
H A Dtpu.py635 op._add_control_input(self.GetControlPivot())
H A Dtensor_tracer.py1996 consumer_op._add_control_input(trace_op)