Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/core/graph/
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 …]
Dgraph_test.cc188 const Edge* source_to_a = graph_.AddControlEdge(graph_.source_node(), a); in TEST_F()
189 graph_.AddControlEdge(a, graph_.sink_node()); in TEST_F()
191 graph_.AddControlEdge(c, graph_.sink_node()); in TEST_F()
299 const Edge* source_to_a = graph_.AddControlEdge(graph_.source_node(), a); in TEST_F()
300 graph_.AddControlEdge(a, graph_.sink_node()); in TEST_F()
302 graph_.AddControlEdge(c, graph_.sink_node()); in TEST_F()
397 graph_.AddControlEdge(graph_.source_node(), a); in TEST_F()
398 graph_.AddControlEdge(b, graph_.sink_node()); in TEST_F()
447 TEST_F(GraphTest, AddControlEdge) { in TEST_F() argument
457 const Edge* edge = graph_.AddControlEdge(c, a); in TEST_F()
[all …]
Dmkl_layout_pass_test.cc559 const Edge* edge = graph_.AddControlEdge(a1, c); in TEST_F()
560 const Edge* edge_1 = graph_.AddControlEdge(a1, e); in TEST_F()
611 const Edge* edge = graph_.AddControlEdge(c, a1); in TEST_F()
612 const Edge* edge_1 = graph_.AddControlEdge(e, a1); in TEST_F()
1487 const Edge* edge = graph_.AddControlEdge(x, c); in TEST_F()
1488 const Edge* edge_1 = graph_.AddControlEdge(x, f); in TEST_F()
1543 const Edge* edge = graph_.AddControlEdge(c, x); in TEST_F()
1544 const Edge* edge_1 = graph_.AddControlEdge(f, x); in TEST_F()
Dsubgraph.cc86 g->AddControlEdge(g->source_node(), feed_node, true); in FeedInputs()
114 g->AddControlEdge(feed_node, e->dst(), true); in FeedInputs()
169 g->AddControlEdge(fetch_node, g->sink_node(), true); in FetchOutputs()
Dalgorithm.cc254 g->AddControlEdge(g->source_node(), n, in FixupSourceAndSinkEdges()
259 g->AddControlEdge(n, g->sink_node(), true /* skip test for duplicates */); in FixupSourceAndSinkEdges()
Dmkl_layout_pass.cc1637 CHECK_NOTNULL((*g)->AddControlEdge(orig_input0, *out, true)); in GetDummyMklTensorNode()
2764 CHECK_NOTNULL((*g)->AddControlEdge(e->src(), new_node, true)); in MergeConv2DWithBiasAdd()
2771 CHECK_NOTNULL((*g)->AddControlEdge(e->src(), new_node, true)); in MergeConv2DWithBiasAdd()
2781 CHECK_NOTNULL((*g)->AddControlEdge(new_node, e->dst(), true)); in MergeConv2DWithBiasAdd()
2790 CHECK_NOTNULL((*g)->AddControlEdge(new_node, e->dst(), true)); in MergeConv2DWithBiasAdd()
2941 (*g)->AddControlEdge(e->src(), new_node, false); in MergePadWithConv2D()
2947 (*g)->AddControlEdge(e->src(), new_node, false); in MergePadWithConv2D()
2956 (*g)->AddControlEdge(new_node, e->dst(), false); in MergePadWithConv2D()
2965 (*g)->AddControlEdge(new_node, e->dst(), false); in MergePadWithConv2D()
3051 CHECK_NOTNULL((*g)->AddControlEdge(e->src(), new_node, true)); in MergeConv2DBackpropFilterWithBiasAddGrad()
[all …]
/external/tensorflow/tensorflow/compiler/jit/
Dresource_operation_safety_analysis_test.cc86 root.graph()->AddControlEdge(write, read); in TEST()
102 root.graph()->AddControlEdge(read, write); in TEST()
128 root.graph()->AddControlEdge(read, modify); in TEST()
142 root.graph()->AddControlEdge(modify, read); in TEST()
158 root.graph()->AddControlEdge(modify, write); in TEST()
172 root.graph()->AddControlEdge(write, modify); in TEST()
189 root.graph()->AddControlEdge(read, modify); in TEST()
190 root.graph()->AddControlEdge(modify, write); in TEST()
205 root.graph()->AddControlEdge(write, modify); in TEST()
206 root.graph()->AddControlEdge(modify, read); in TEST()
[all …]
Dbuild_xla_ops_pass.cc64 scope.graph()->AddControlEdge(control, data.node()); in ControlToData()
124 s.graph()->AddControlEdge(new_node, s.graph()->sink_node()); in MergeOutgoingControlEdges()
144 s.graph()->AddControlEdge(ctrl_merge.node(), e->dst()); in MergeOutgoingControlEdges()
201 g->AddControlEdge(e->src(), to); in CopyIncomingControlEdges()
254 g->AddControlEdge(e->src(), call.operation.node()); in ReplaceFunctionCallWithPartionedCall()
263 g->AddControlEdge(call.operation.node(), e->dst()); in ReplaceFunctionCallWithPartionedCall()
389 g->AddControlEdge( in ReplaceNodeWithXlaCompileAndXlaRun()
Dmark_for_compilation_pass_test.cc552 root.graph()->AddControlEdge(read, write); in TEST()
572 root.graph()->AddControlEdge(write, read); in TEST()
593 root.graph()->AddControlEdge(write_0, neutral_0); in TEST()
594 root.graph()->AddControlEdge(neutral_0, read_0); in TEST()
595 root.graph()->AddControlEdge(read_0, write_1); in TEST()
596 root.graph()->AddControlEdge(write_1, neutral_1); in TEST()
597 root.graph()->AddControlEdge(neutral_1, read_1); in TEST()
633 graph->AddControlEdge(a, b); in TEST()
634 graph->AddControlEdge(b, c); in TEST()
635 graph->AddControlEdge(c, a); in TEST()
[all …]
Ddeadness_analysis_test.cc139 root.graph()->AddControlEdge(iv.output.node(), increment_by.node()); in CreateInductionVariable()
140 root.graph()->AddControlEdge(iv.output.node(), final_value.node()); in CreateInductionVariable()
789 root.graph()->AddControlEdge(id0.node(), const0.node()); in TEST()
790 root.graph()->AddControlEdge(id1.node(), const1.node()); in TEST()
813 root.graph()->AddControlEdge(id0.node(), ctrl_trigger0.operation.node()); in TEST()
814 root.graph()->AddControlEdge(ctrl_trigger0.operation.node(), const0.node()); in TEST()
816 root.graph()->AddControlEdge(id1.node(), ctrl_trigger1.operation.node()); in TEST()
817 root.graph()->AddControlEdge(ctrl_trigger1.operation.node(), const1.node()); in TEST()
837 root.graph()->AddControlEdge(id0.node(), m0.output.node()); in TEST()
838 root.graph()->AddControlEdge(id1.node(), m1.output.node()); in TEST()
Dclone_constants_for_better_clustering.cc47 g->AddControlEdge(e->src(), new_in); in CloneNode()
108 g->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()
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()
Dnode_matchers_test.cc110 root.graph()->AddControlEdge(placeholder_a.node(), placeholder_c.node()); in TEST()
111 root.graph()->AddControlEdge(placeholder_b.node(), placeholder_c.node()); in TEST()
Dextract_outside_compilation_pass.cc143 g->AddControlEdge(recv_at_host_node, edge.dst); in ReplaceArgNodesWithRecvAtHostNode()
251 g->AddControlEdge(edge->src(), send_from_host_node); in ReplaceRetNodesWithSendFromHostNode()
555 host_graph.AddControlEdge(copy, sequencer); in ConstructHostGraph()
574 host_graph.AddControlEdge(sequencer, host_graph.sink_node()); in ConstructHostGraph()
670 main_graph->AddControlEdge(copy, xla_computation_node); in ExpandHostGraphIntoMainGraph()
1410 g->AddControlEdge(send_pred_node, n); in ExtractOutsideCompilationForNodesWithAssociatedFunctions()
1536 (*graph)->AddControlEdge(n, send_from_host_node); in operator ()()
1540 (*graph)->AddControlEdge(recv_at_host_node, n); in operator ()()
1556 (*graph)->AddControlEdge(send_from_host_node, (*graph)->sink_node()); in operator ()()
Dbuild_xla_ops_pass_test.cc143 root.graph()->AddControlEdge(call, write_op); in TEST_F()
165 root.graph()->AddControlEdge(call, write_op); in TEST_F()
/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/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/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()
/external/tensorflow/tensorflow/core/common_runtime/
Daccumulate_n_optimizer.cc176 g->AddControlEdge(control_edge->src(), initialize_accumulator); in rewriteNode()
180 g->AddControlEdge(assign_add, clean_up_accumulator); in rewriteNode()
185 g->AddControlEdge(clean_up_accumulator, out_edge->dst()); in rewriteNode()
Dlower_if_op.cc183 graph_->AddControlEdge(e->src(), control_predecessor_); in AddInputs()
192 graph_->AddControlEdge(pivot_t_, then_call_node_); in AddOutputs()
194 graph_->AddControlEdge(pivot_f_, else_call_node_); in AddOutputs()
228 graph_->AddControlEdge(lowered_if_output_, e->dst()); in AddOutputs()
Dparallel_concat_optimizer.cc85 g->AddControlEdge(input_edge->src(), start); in Run()
113 g->AddControlEdge(identity_node, e->dst()); in Run()
Dlower_while_op.cc223 graph_->AddControlEdge(incoming_control_node, n); in CreateEnterNodes()
252 graph_->AddControlEdge(merge_nodes_[0], cond_call_node_); in CreateCondFuncCallNode()
308 graph_->AddControlEdge(body_control_node_, body_call_node_); in CreateBodyFuncCallNode()
361 graph_->AddControlEdge(lowered_while_output_, e->dst()); in UpdateConsumers()
Dfunction.cc1303 g->AddControlEdge(in->src(), out->dst()); in RemoveIdentityNodes()
1352 g->AddControlEdge(e->src(), input_control_node); in RemoveListArrayConverter()
1370 g->AddControlEdge(input_control_node, id); in RemoveListArrayConverter()
1384 g->AddControlEdge(output_control_node, e->dst()); in RemoveListArrayConverter()
1402 g->AddControlEdge(id, output_control_node); in RemoveListArrayConverter()
1627 g->AddControlEdge(e->src(), input_control_node); in InlineFunctionBody()
1682 g->AddControlEdge(input_control_node, clone); in InlineFunctionBody()
1709 g->AddControlEdge(input_control_node, n); in InlineFunctionBody()
1713 g->AddControlEdge(n, e->dst()); in InlineFunctionBody()
1752 g->AddControlEdge(e->src(), n); in InlineFunctionBody()
[all …]
/external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/
Dsegment.cc119 void AddControlEdge(SimpleNode* src, SimpleNode* dst);
200 void SimpleGraph::AddControlEdge(SimpleNode* src, SimpleNode* dst) { in AddControlEdge() function in tensorflow::tensorrt::segment::SimpleGraph
361 graph->AddControlEdge(e->src(), src); in ContractEdge()
380 graph->AddControlEdge(src, e->dst()); in ContractEdge()

123