Searched refs:control_predecessor (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | call_inliner.cc | 56 for (HloInstruction* control_predecessor : hlo->control_predecessors()) { in DefaultAction()
58 Resolve(control_predecessor)); in DefaultAction()
|
| D | hlo_module_group_util.cc | 94 for (HloInstruction* control_predecessor : hlo->control_predecessors()) { in GlobalPredecessors()
95 add_unique_predecessor(control_predecessor); in GlobalPredecessors()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.h | 376 Node* control_predecessor, Graph* graph,
439 Node* control_predecessor, Node* control_successor,
|
| D | distributed_tpu_rewrite_pass.cc | 664 Node* control_predecessor, Node* split_node, in CreatePadNode() argument
706 graph->AddControlEdge(control_predecessor, pad_node); in CreatePadNode()
716 Node* control_predecessor, Node* orig_src, in CreateSplitNode() argument
724 CreatePadNode(padding, num_dims, dim, dtype, control_predecessor, in CreateSplitNode()
773 graph->AddControlEdge(control_predecessor, split_dim_node); in CreateSplitNode()
794 int orig_src_output, Node* orig_src, Node* control_predecessor, in CreateOrGetSplitNodesForInputSharding() argument
842 control_predecessor, orig_src, graph)); in CreateOrGetSplitNodesForInputSharding()
866 control_predecessor, input_split_node, graph)); in CreateOrGetSplitNodesForInputSharding()
2582 absl::Span<const VariableInput> variables, Node* control_predecessor, in BuildVariableReads() argument
2607 graph->AddControlEdge(control_predecessor, read_node); in BuildVariableReads()
[all …]
|