Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/graph/
Dquantize_training_test.cc89 g->AddControlEdge(g->source_node(), a); in TEST_F()
90 g->AddControlEdge(g->source_node(), b); in TEST_F()
94 g->AddControlEdge(m1, g->sink_node()); in TEST_F()
140 g->AddControlEdge(g->source_node(), a); in TEST_F()
141 g->AddControlEdge(g->source_node(), b); in TEST_F()
145 g->AddControlEdge(m1, g->sink_node()); in TEST_F()
187 g->AddControlEdge(g->source_node(), a); in TEST_F()
188 g->AddControlEdge(g->source_node(), b); in TEST_F()
189 g->AddControlEdge(g->source_node(), c); in TEST_F()
190 g->AddControlEdge(g->source_node(), d); in TEST_F()
[all …]
Dgraph_test.cc186 const Edge* source_to_a = graph_.AddControlEdge(graph_.source_node(), a); in TEST_F()
187 graph_.AddControlEdge(a, graph_.sink_node()); in TEST_F()
189 graph_.AddControlEdge(c, graph_.sink_node()); in TEST_F()
297 const Edge* source_to_a = graph_.AddControlEdge(graph_.source_node(), a); in TEST_F()
298 graph_.AddControlEdge(a, graph_.sink_node()); in TEST_F()
300 graph_.AddControlEdge(c, graph_.sink_node()); in TEST_F()
395 graph_.AddControlEdge(graph_.source_node(), a); in TEST_F()
396 graph_.AddControlEdge(b, graph_.sink_node()); in TEST_F()
445 TEST_F(GraphTest, AddControlEdge) { in TEST_F() argument
455 const Edge* edge = graph_.AddControlEdge(c, a); in TEST_F()
[all …]
Dsubgraph.cc108 g->AddControlEdge(g->source_node(), recv_node); in FeedInputs()
134 g->AddControlEdge(recv_node, e->dst()); in FeedInputs()
257 g->AddControlEdge(send_node, g->sink_node()); in FetchOutputs()
Dmkl_layout_pass.cc1059 CHECK_NOTNULL((*g)->AddControlEdge(orig_input0, *out)); in GetDummyMklTensorNode()
1352 CHECK_NOTNULL((*g)->AddControlEdge(orig_input0, *out)); in GetDummyWorkspaceTensorNode()
1842 CHECK_NOTNULL((*g)->AddControlEdge(e->src(), new_node)); in MergeNode()
1847 CHECK_NOTNULL((*g)->AddControlEdge(e->src(), new_node)); in MergeNode()
1858 CHECK_NOTNULL((*g)->AddControlEdge(new_node, e->dst())); in MergeNode()
1865 CHECK_NOTNULL((*g)->AddControlEdge(new_node, e->dst())); in MergeNode()
1995 CHECK_NOTNULL((*g)->AddControlEdge(e->src(), new_node)); in RewriteNode()
2008 CHECK_NOTNULL((*g)->AddControlEdge(new_node, e->dst())); in RewriteNode()
3111 CHECK_NOTNULL((*g)->AddControlEdge(orig_input0, *out, true));
3407 CHECK_NOTNULL((*g)->AddControlEdge(orig_input0, *out, true));
[all …]
Dalgorithm.cc220 g->AddControlEdge(g->source_node(), n); in FixupSourceAndSinkEdges()
224 g->AddControlEdge(n, g->sink_node()); in FixupSourceAndSinkEdges()
Dnode_builder.cc124 graph->AddControlEdge(control_input, node); in Finalize()
Dquantize_training.cc238 graph->AddControlEdge(new_save_op, edge->dst()); in ConnectVariablesToSaveOp()
304 graph->AddControlEdge(assign_op, restore_all); in AddRestoreVariableSubgraphs()
Dgraph.cc291 AddControlEdge(source, sink); in Graph()
427 const Edge* Graph::AddControlEdge(Node* source, Node* dest, in AddControlEdge() function in tensorflow::Graph
Dmkl_tfconversion_pass.cc249 CHECK_NOTNULL((*g)->AddControlEdge(e->src(), conversion_node)); in InsertInputConversionNode()
Dgraph.h446 const Edge* AddControlEdge(Node* source, Node* dest,
Dgraph_constructor.cc1006 g_->AddControlEdge(inputs[i].node, node); in Convert()
1036 g_->AddControlEdge(src_node, e.dst_node); in AddBackEdges()
/external/tensorflow/tensorflow/contrib/nccl/kernels/
Dnccl_rewrite.cc81 graph->AddControlEdge(send_node, out_node.node); in ReplaceReduce()
94 graph->AddControlEdge(recv_node, out_node.node); in ReplaceReduce()
150 graph->AddControlEdge(in_node, out_node.node); in ReplaceBroadcast()
191 graph->AddControlEdge(send_node, out_node.node); in ReplaceBroadcast()
196 graph->AddControlEdge(send_node, out_node.node); in ReplaceBroadcast()
/external/tensorflow/tensorflow/core/common_runtime/
Daccumulate_n_optimizer.cc160 g->AddControlEdge(control_edge->src(), initialize_accumulator); in rewriteNode()
164 g->AddControlEdge(assign_add, clean_up_accumulator); in rewriteNode()
169 g->AddControlEdge(clean_up_accumulator, out_edge->dst()); in rewriteNode()
Dparallel_concat_optimizer.cc84 g->AddControlEdge(input_edge->src(), start); in Run()
112 g->AddControlEdge(identity_node, e->dst()); in Run()
Dfunction.cc1028 g->AddControlEdge(in->src(), out->dst()); in RemoveIdentityNodes()
1069 g->AddControlEdge(e->src(), input_control_node); in RemoveListArrayConverter()
1087 g->AddControlEdge(input_control_node, id); in RemoveListArrayConverter()
1101 g->AddControlEdge(output_control_node, e->dst()); in RemoveListArrayConverter()
1119 g->AddControlEdge(id, output_control_node); in RemoveListArrayConverter()
1176 g->AddControlEdge(e->src(), input_control_node); in InlineFunctionBody()
1220 g->AddControlEdge(input_control_node, clone); in InlineFunctionBody()
1246 g->AddControlEdge(input_control_node, n); in InlineFunctionBody()
1250 g->AddControlEdge(n, e->dst()); in InlineFunctionBody()
1287 g->AddControlEdge(e->src(), n); in InlineFunctionBody()
[all …]
Dconstant_folding.cc539 graph->AddControlEdge(graph->source_node(), constant_node); in ReplaceTensorWithConstant()
542 graph->AddControlEdge(node, constant_node); in ReplaceTensorWithConstant()
Dmemory_types.cc193 g->AddControlEdge(send, recv); in EnsureMemoryTypes()
Dfunction_test.cc647 s.graph()->AddControlEdge(c.operation.node(), b.node()); in TEST_F()
663 s.graph()->AddControlEdge(func0.operation.node(), b_x2.node()); in TEST_F()
665 s.graph()->AddControlEdge(func0.operation.node(), b_y.node()); in TEST_F()
/external/tensorflow/tensorflow/compiler/jit/
Dbuild_xla_launch_ops_pass.cc93 graph->AddControlEdge(edge->src(), launch_node); in ReplaceNodeWithXlaLaunch()
110 graph->AddControlEdge(launch_node, dst); in ReplaceNodeWithXlaLaunch()
Dencapsulate_subgraphs_pass.cc814 graph_->AddControlEdge(src_image, host_compute); in AddHostComputes()
831 graph_->AddControlEdge(host_compute, dst_image); in AddHostComputes()
862 graph_out->AddControlEdge(sequencer_, node); in ConnectSequencerToOutputs()
1095 graph_out->AddControlEdge(oc_subgraph->recv_at_host, sequencer_); in AddRecvAtHostNode()
1138 graph_out->AddControlEdge(oc_subgraph->send_from_host, sequencer_); in AddSendFromHostNode()
1242 g->AddControlEdge(src_image, dst_image); in CopySubgraphEdges()
1530 graph_out->AddControlEdge(src_image, dst_image); in CopyEdgeToOutputGraph()
1536 graph_out->AddControlEdge(src_image, node_images.at(edge->dst())); in CopyEdgeToOutputGraph()
1595 graph_out->AddControlEdge(src_image, dst_image); in AddEdgesToOutputGraph()
Dmark_for_compilation_pass_test.cc513 graph->AddControlEdge(a, b); in TEST()
514 graph->AddControlEdge(b, c); in TEST()
515 graph->AddControlEdge(c, a); in TEST()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dexecutor_test.cc257 g->AddControlEdge(init, add); // Ensures run after init. in BuildConcurrentAddAssign()
259 g->AddControlEdge(assign, out); in BuildConcurrentAddAssign()
383 g->AddControlEdge(init, two); // Ensures run after init. in TEST_F()
/external/tensorflow/tensorflow/core/kernels/
Dconstant_op_test.cc104 g->AddControlEdge(prev, curr); in ManyConsts()
/external/tensorflow/tensorflow/c/
Dpython_api.cc24 graph->graph.AddControlEdge(&input->node, &op->node); in AddControlInput()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_control_flow.cc521 graph->AddControlEdge(in_edge->src(), while_node); in FunctionalizeLoop()
537 graph->AddControlEdge(while_node, dst); in FunctionalizeLoop()
1126 graph_->AddControlEdge(old, if_node); in FunctionalizeInternal()
1277 graph_->AddControlEdge(in_edge->src(), if_node); in AddInputEdges()

12