Searched refs:control_successor (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | cond_builder.h | 54 Node* control_successor();
|
| D | cond_builder.cc | 71 Node* CondBuilder::control_successor() { return control_successor_; } in control_successor() function in tensorflow::CondBuilder
|
| D | distributed_tpu_rewrite_pass.h | 400 absl::Span<const VariableInput> variables, Node* control_successor,
439 Node* control_predecessor, Node* control_successor,
|
| D | distributed_tpu_rewrite_pass.cc | 2635 absl::Span<const VariableInput> variables, Node* control_successor, in BuildVariableWrites() argument
2678 graph->AddControlEdge(cb.control_successor(), control_successor); in BuildVariableWrites()
2896 Node* control_predecessor, Node* control_successor, in BuildExecuteNodes() argument
3111 graph->AddControlEdge(node, control_successor); in BuildExecuteNodes()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_module_group_util.cc | 172 for (HloInstruction* control_successor : hlo->control_successors()) { in GlobalSuccessors()
173 add_unique_successor(control_successor); in GlobalSuccessors()
|