Home
last modified time | relevance | path

Searched refs:AddControlEdge (Results 1 – 25 of 76) sorted by relevance

1234

/external/tensorflow/tensorflow/core/common_runtime/
Dquantize_training_test.cc90 g->AddControlEdge(g->source_node(), a); in TEST_F()
91 g->AddControlEdge(g->source_node(), b); in TEST_F()
95 g->AddControlEdge(m1, g->sink_node()); in TEST_F()
141 g->AddControlEdge(g->source_node(), a); in TEST_F()
142 g->AddControlEdge(g->source_node(), b); in TEST_F()
146 g->AddControlEdge(m1, g->sink_node()); in TEST_F()
188 g->AddControlEdge(g->source_node(), a); in TEST_F()
189 g->AddControlEdge(g->source_node(), b); in TEST_F()
190 g->AddControlEdge(g->source_node(), c); in TEST_F()
191 g->AddControlEdge(g->source_node(), d); in TEST_F()
[all …]
Dfunction_utils.cc150 g->AddControlEdge(in->src(), out->dst()); in RemoveIdentityNodes()
201 g->AddControlEdge(e->src(), input_control_node); in RemoveListArrayConverter()
219 g->AddControlEdge(input_control_node, id); in RemoveListArrayConverter()
233 g->AddControlEdge(output_control_node, e->dst()); in RemoveListArrayConverter()
251 g->AddControlEdge(id, output_control_node); in RemoveListArrayConverter()
Daccumulate_n_optimizer.cc208 g->AddControlEdge(control_edge->src(), initialize_accumulator); in RewriteIntoTempVariable()
212 g->AddControlEdge(assign_add, clean_up_accumulator); in RewriteIntoTempVariable()
217 g->AddControlEdge(clean_up_accumulator, out_edge->dst()); in RewriteIntoTempVariable()
273 g->AddControlEdge(add_n_node, out_edge->dst()); in RewriteIntoAddN()
Dinline_function_utils.cc568 g->AddControlEdge(e->src(), input_control_node, kDoNotCheckDuplicates); in InlineFunctionBody()
673 g->AddControlEdge(input_control_node, clone, kDoNotCheckDuplicates); in InlineFunctionBody()
706 g->AddControlEdge(input_control_node, n, kDoNotCheckDuplicates); in InlineFunctionBody()
710 g->AddControlEdge(n, e->dst(), kDoNotCheckDuplicates); in InlineFunctionBody()
758 g->AddControlEdge(e->src(), n, kDoNotCheckDuplicates); in InlineFunctionBody()
779 g->AddControlEdge(n, output_control_node, kDoNotCheckDuplicates); in InlineFunctionBody()
785 g->AddControlEdge(n, output_control_node, kDoNotCheckDuplicates); in InlineFunctionBody()
801 g->AddControlEdge(input_control_node, output_control_node, in InlineFunctionBody()
811 g->AddControlEdge(output_control_node, e->dst(), kDoNotCheckDuplicates); in InlineFunctionBody()
Dlower_if_op.cc220 graph_->AddControlEdge(e->src(), control_predecessor_); in AddInputs()
231 graph_->AddControlEdge(pivot_t_, then_call_node_); in AddOutputs()
233 graph_->AddControlEdge(pivot_f_, else_call_node_); in AddOutputs()
271 graph_->AddControlEdge(branch_executed_node_, e->dst()); in AddOutputs()
Dparallel_concat_optimizer.cc80 g->AddControlEdge(input_edge->src(), start); in Run()
106 g->AddControlEdge(identity_node, e->dst()); in Run()
Dlower_case_op.cc184 graph_->AddControlEdge(e->src(), control_predecessor_); in AddInputs()
196 graph_->AddControlEdge(pivots_[b], call_nodes_[b]); in AddOutputs()
243 graph_->AddControlEdge(branch_executed_node_, e->dst()); in AddOutputs()
Dlower_while_op.cc275 graph_->AddControlEdge(incoming_control_node, n); in CreateEnterNodes()
312 graph_->AddControlEdge(merge_nodes_[0], cond_call_node_); in CreateCondFuncCallNode()
378 graph_->AddControlEdge(body_control_node_, body_call_node_); in CreateBodyFuncCallNode()
473 graph_->AddControlEdge(lowered_while_executed_, e->dst()); in UpdateConsumers()
Dsingle_threaded_executor_test.cc135 g->AddControlEdge(arg, ret); in TestContext()
210 g->AddControlEdge(in1, ret); in TEST_F()
347 g->AddControlEdge(in0, add); in TEST_F()
348 g->AddControlEdge(one, ret); in TEST_F()
/external/tensorflow/tensorflow/compiler/jit/
Dresource_operation_safety_analysis_test.cc85 root.graph()->AddControlEdge(write, read); in TEST()
101 root.graph()->AddControlEdge(read, write); in TEST()
127 root.graph()->AddControlEdge(read, modify); in TEST()
141 root.graph()->AddControlEdge(modify, read); in TEST()
157 root.graph()->AddControlEdge(modify, write); in TEST()
171 root.graph()->AddControlEdge(write, modify); in TEST()
188 root.graph()->AddControlEdge(read, modify); in TEST()
189 root.graph()->AddControlEdge(modify, write); in TEST()
204 root.graph()->AddControlEdge(write, modify); in TEST()
205 root.graph()->AddControlEdge(modify, read); in TEST()
[all …]
Dbuild_xla_ops_pass.cc103 scope.graph()->AddControlEdge(control, data.node()); in ControlToData()
192 s.graph()->AddControlEdge(new_node, s.graph()->sink_node()); in MergeOutgoingControlEdges()
212 s.graph()->AddControlEdge(ctrl_merge.node(), e->dst()); in MergeOutgoingControlEdges()
269 g->AddControlEdge(e->src(), to); in CopyIncomingControlEdges()
322 g->AddControlEdge(e->src(), call.operation.node()); in ReplaceFunctionCallWithPartitionedCall()
331 g->AddControlEdge(call.operation.node(), e->dst()); in ReplaceFunctionCallWithPartitionedCall()
455 root.graph()->AddControlEdge(predicate_as_control.node(), in PredicateInt32Inputs()
556 g->AddControlEdge(inverse_predicate_as_control.node(), n); in ReplaceNodeWithXlaCompileAndXlaRun()
Dclone_constants_for_better_clustering.cc67 graph_->AddControlEdge(e->src(), new_in); in CloneNode()
145 graph_->AddControlEdge(input_cloned, e->dst()); in CloneSmallHostConstantInputs()
Dincrease_dynamism_for_auto_jit_pass.cc159 scope_.graph()->AddControlEdge(e->src(), new_const.node()); in Get1DHostConstant()
225 host_scope.graph()->AddControlEdge(e->src(), concat_axis.node()); in ComputeSliceSize()
298 g->AddControlEdge(e->src(), static_shaped_slice); in ReplaceTensorFlowSliceWithStaticShapedSlice()
Dxla_cluster_util_test.cc176 s.graph()->AddControlEdge(constant.node(), variable.node()); in CreateSubgraphTouchingRefVar()
193 s.graph()->AddControlEdge(call.operation.node(), constant.node()); in CreateSubgraphCallingFunctionWithRefVar()
203 s.graph()->AddControlEdge(call.operation.node(), constant.node()); in CreateSubgraphCallingFunctionWithoutRefVar()
Dencapsulate_xla_computations_pass_test.cc170 graph->AddControlEdge(get_node_in_graph(a0.node()), in TEST()
172 graph->AddControlEdge(get_node_in_graph(a1.node()), in TEST()
175 graph->AddControlEdge(get_node_in_graph(a1.node()), in TEST()
177 graph->AddControlEdge(get_node_in_graph(a0.node()), in TEST()
Dmark_for_compilation_pass_test.cc724 root.graph()->AddControlEdge(read, write); in TEST()
744 root.graph()->AddControlEdge(write, read); in TEST()
765 root.graph()->AddControlEdge(write_0, neutral_0); in TEST()
766 root.graph()->AddControlEdge(neutral_0, read_0); in TEST()
767 root.graph()->AddControlEdge(read_0, write_1); in TEST()
768 root.graph()->AddControlEdge(write_1, neutral_1); in TEST()
769 root.graph()->AddControlEdge(neutral_1, read_1); in TEST()
805 graph->AddControlEdge(a, b); in TEST()
806 graph->AddControlEdge(b, c); in TEST()
807 graph->AddControlEdge(c, a); in TEST()
[all …]
/external/tensorflow/tensorflow/core/graph/
Dgraph_test.cc189 const Edge* source_to_a = graph_.AddControlEdge(graph_.source_node(), a); in TEST_F()
190 graph_.AddControlEdge(a, graph_.sink_node()); in TEST_F()
192 graph_.AddControlEdge(c, graph_.sink_node()); in TEST_F()
300 const Edge* source_to_a = graph_.AddControlEdge(graph_.source_node(), a); in TEST_F()
301 graph_.AddControlEdge(a, graph_.sink_node()); in TEST_F()
303 graph_.AddControlEdge(c, graph_.sink_node()); in TEST_F()
398 graph_.AddControlEdge(graph_.source_node(), a); in TEST_F()
399 graph_.AddControlEdge(b, graph_.sink_node()); in TEST_F()
448 TEST_F(GraphTest, AddControlEdge) { in TEST_F() argument
458 const Edge* edge = graph_.AddControlEdge(c, a); in TEST_F()
[all …]
Dsubgraph.cc85 g->AddControlEdge(g->source_node(), feed_node, true); in FeedInputs()
113 g->AddControlEdge(feed_node, e->dst(), true); in FeedInputs()
168 g->AddControlEdge(fetch_node, g->sink_node(), true); in FetchOutputs()
/external/angle/third_party/vulkan-deps/spirv-tools/src/source/opt/
Dpropagator.cpp20 void SSAPropagator::AddControlEdge(const Edge& edge) { in AddControlEdge() function in spvtools::opt::SSAPropagator
112 AddControlEdge(e); in Simulate()
126 AddControlEdge(Edge(ctx_->get_instr_block(instr), dest_bb)); in Simulate()
202 AddControlEdge(bb_succs_.at(block).at(0)); in Simulate()
235 AddControlEdge(e); in Initialize()
/external/deqp-deps/SPIRV-Tools/source/opt/
Dpropagator.cpp20 void SSAPropagator::AddControlEdge(const Edge& edge) { in AddControlEdge() function in spvtools::opt::SSAPropagator
112 AddControlEdge(e); in Simulate()
126 AddControlEdge(Edge(ctx_->get_instr_block(instr), dest_bb)); in Simulate()
202 AddControlEdge(bb_succs_.at(block).at(0)); in Simulate()
235 AddControlEdge(e); in Initialize()
/external/swiftshader/third_party/SPIRV-Tools/source/opt/
Dpropagator.cpp20 void SSAPropagator::AddControlEdge(const Edge& edge) { in AddControlEdge() function in spvtools::opt::SSAPropagator
112 AddControlEdge(e); in Simulate()
126 AddControlEdge(Edge(ctx_->get_instr_block(instr), dest_bb)); in Simulate()
202 AddControlEdge(bb_succs_.at(block).at(0)); in Simulate()
235 AddControlEdge(e); in Initialize()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dhost_training_loop_optimization_util.cc386 graph->AddControlEdge(after_switch_node, after_last_iteration_node); in AddNoOpAfterLastIteration()
390 graph->AddControlEdge(after_last_iteration_node, out_node); in AddNoOpAfterLastIteration()
531 graph->AddControlEdge(loop_condition_node, default_sharding_node); in AddReshardOp()
558 graph->AddControlEdge(before_loop_iteration_node, reshard_op_node); in AddReshardOp()
559 graph->AddControlEdge(reshard_op_node, execute_node); in AddReshardOp()
590 graph->AddControlEdge(loop_condition_node, var_handle_node); in AddReshardOp()
623 graph->AddControlEdge(after_last_iteration_node, unshard_op_node); in AddReshardOp()
624 graph->AddControlEdge(unshard_op_node, after_unshard_node); in AddReshardOp()
635 graph->AddControlEdge(after_unshard_node, exit); in AddReshardOp()
Ddistributed_tpu_configuration_rewrite_pass.cc154 graph->AddControlEdge(node, sync_node); in AddSynchronizationNode()
162 graph->AddControlEdge(sync_node, dep.dst); in AddSynchronizationNode()
190 graph->AddControlEdge(*shutdown_node, dep.dst); in AddShutdownNode()
211 graph->AddControlEdge(src_node, host_disconnect_node); in AddHostDisconnectNode()
214 graph->AddControlEdge(host_disconnect_node, post_disconnect_node); in AddHostDisconnectNode()
Dconfigure_tpu_embedding_rewrite_pass.cc64 graph->AddControlEdge(end_node, sync_node); in AddSynchronizationNode()
71 graph->AddControlEdge(sync_node, dep.dst); in AddSynchronizationNode()
116 graph->AddControlEdge(src_node, *partitioner_node); in AddExecutePartitionerNode()
/external/tensorflow/tensorflow/core/nccl/
Dnccl_rewrite.cc83 graph->AddControlEdge(send_node, out_node.node); in ReplaceReduce()
96 graph->AddControlEdge(recv_node, out_node.node); in ReplaceReduce()
152 graph->AddControlEdge(in_node, out_node.node); in ReplaceBroadcast()
193 graph->AddControlEdge(send_node, out_node.node); in ReplaceBroadcast()
198 graph->AddControlEdge(send_node, out_node.node); in ReplaceBroadcast()

1234