Searched refs:control_before (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.h | 537 Graph* graph, Node* host_transfer_sequencer, Node* control_before,
546 Node** control_before,
|
| D | distributed_tpu_rewrite_pass.cc | 4048 Graph* graph, Node** host_transfer_sequencer, Node** control_before, in BuildSequencingNodes() argument
4056 /*device=*/"", graph, control_before)); in BuildSequencingNodes()
4077 graph->AddControlEdge(predecessor, *control_before); in BuildSequencingNodes()
4098 Graph* graph, Node* host_transfer_sequencer, Node* control_before, in DealWithConstantsAndVariables() argument
4105 TF_RETURN_IF_ERROR(BuildVariableReads(variable_nodes, control_before, graph, in DealWithConstantsAndVariables()
4294 Node* control_before; in RewriteTPUReplicateNode() local
4298 &control_before, &control_after)); in RewriteTPUReplicateNode()
4309 control_before, control_after, variable_inputs, in RewriteTPUReplicateNode()
4330 graph->AddControlEdge(control_before, compile_node); in RewriteTPUReplicateNode()
|