Home
last modified time | relevance | path

Searched refs:output_control_node (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dinline_function_utils.cc764 Node* output_control_node = nullptr; in InlineFunctionBody() local
774 output_control_node = no_op("output_control_node"); in InlineFunctionBody()
775 VLOG(4) << "Add output control node: " << output_control_node->name(); in InlineFunctionBody()
779 g->AddControlEdge(n, output_control_node, kDoNotCheckDuplicates); in InlineFunctionBody()
785 g->AddControlEdge(n, output_control_node, kDoNotCheckDuplicates); in InlineFunctionBody()
796 if (output_control_node && output_control_node->in_edges().empty()) { in InlineFunctionBody()
800 << output_control_node->name(); in InlineFunctionBody()
801 g->AddControlEdge(input_control_node, output_control_node, in InlineFunctionBody()
811 g->AddControlEdge(output_control_node, e->dst(), kDoNotCheckDuplicates); in InlineFunctionBody()
833 .ControlInput(output_control_node) in InlineFunctionBody()
[all …]
H A Dfunction_utils.cc223 Node* output_control_node = nullptr; in RemoveListArrayConverter() local
226 if (output_control_node == nullptr) { in RemoveListArrayConverter()
231 output_control_node = no_op("output_control_node"); in RemoveListArrayConverter()
233 g->AddControlEdge(output_control_node, e->dst()); in RemoveListArrayConverter()
249 if (output_control_node != nullptr) { in RemoveListArrayConverter()
251 g->AddControlEdge(id, output_control_node); in RemoveListArrayConverter()
H A Dfunction_test.cc1051 const string output_control_node = "Func/b/output_control_node/_2"; in TEST_F() local
1065 NDef(output_control_node, "NoOp", {"^Func/b/output/_1"}, {}), in TEST_F()
1066 NDef("c", "NoOp", {"^" + output_control_node}, {}), in TEST_F()
1088 NDef(output_control_node, "NoOp", {"^b/add"}, {}), in TEST_F()
1089 NDef("c", "NoOp", {"^" + output_control_node}, {}), in TEST_F()
1126 const string output_control_node = "Func/b/output_control_node/_2"; in TEST_F() local
1137 NDef(output_control_node, "NoOp", {"^b/add"}, {}), in TEST_F()
1157 {output_node, "^" + output_control_node}, in TEST_F()
1175 NDef("b", "NoOp", {"^" + output_control_node}, {})); in TEST_F()
1214 const string output_control_node = "Func/b/output_control_node/_2"; in TEST_F() local
[all …]