| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | lower_while_op_test.cc | 68 Node* while_node; in TEST() local
82 .Finalize(root.graph(), &while_node)); in TEST()
84 root.WithOpName("C").WithControlDependencies(Output(while_node)), in TEST()
85 Output(while_node)); in TEST()
86 TF_ASSERT_OK(root.DoShapeInference(while_node)); in TEST()
160 TF_ASSERT_OK(session.Run(feeds, {Output(while_node)}, &out_tensors)); in TEST()
168 TF_ASSERT_OK(session.Run(feeds, {Output(while_node)}, &out_tensors)); in TEST()
190 Node* while_node; in TEST() local
204 .Finalize(graph.get(), &while_node)); in TEST()
295 Node* while_node; in TEST() local
[all …]
|
| D | lower_functional_ops_test.cc | 157 Node* while_node; in TEST() local
168 .Finalize(root.graph(), &while_node)); in TEST()
169 TF_ASSERT_OK(root.DoShapeInference(while_node)); in TEST()
187 TF_ASSERT_OK(session.Run(feeds, {Output(while_node, 2)}, &out_tensors)); in TEST()
197 TF_ASSERT_OK(session.Run(feeds, {Output(while_node, 2)}, &out_tensors)); in TEST()
|
| D | executor_test.cc | 613 Node* while_node; in BM_WhileLoopHelper() local
634 .Finalize(root.graph(), &while_node)); in BM_WhileLoopHelper()
636 root.WithOpName("C").WithControlDependencies(Output(while_node)), in BM_WhileLoopHelper()
637 Output(while_node)); in BM_WhileLoopHelper()
638 TF_ASSERT_OK(root.DoShapeInference(while_node)); in BM_WhileLoopHelper()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_while.cc | 463 TF_ASSIGN_OR_RETURN(Node * while_node, AddNodeDefToGraph(while_def, graph)); in FunctionalizeLoop()
471 graph->AddControlEdge(in_edge->src(), while_node); in FunctionalizeLoop()
473 graph->AddEdge(in_edge->src(), in_edge->src_output(), while_node, i); in FunctionalizeLoop()
487 graph->AddControlEdge(while_node, dst); in FunctionalizeLoop()
489 graph->AddEdge(while_node, i, dst, dst_input); in FunctionalizeLoop()
503 frame->parent->nodes.insert(while_node); in FunctionalizeLoop()
|
| D | tf2xla_util.cc | 251 Status PropagateConstIntoWhileNode(Graph* g, Node* while_node, in PropagateConstIntoWhileNode() argument
259 TF_RETURN_IF_ERROR(GetNodeAttr(while_node->def(), "body", &body_attr)); in PropagateConstIntoWhileNode()
263 " for While node ", while_node->name()); in PropagateConstIntoWhileNode()
265 for (int i = 0; i < while_node->num_inputs(); i++) { in PropagateConstIntoWhileNode()
267 TF_RETURN_IF_ERROR(while_node->input_node(i, &input_node)); in PropagateConstIntoWhileNode()
302 while_node, attr_name, const_input_index_to_node, lookup_fld, fld)); in PropagateConstIntoWhileNode()
|
| D | tf2xla_util_test.cc | 407 Node* while_node = node_name_index["while"]; in TEST() local
408 ASSERT_NE(while_node, nullptr); in TEST()
409 TF_ASSERT_OK(GetNodeAttr(while_node->def(), "body", &body_fn)); in TEST()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | rearrange_function_argument_pass_test.cc | 154 const Node *while_node = node_name_index.at("while"); in TEST() local
155 ASSERT_NE(while_node, nullptr); in TEST()
156 TF_CHECK_OK(while_node->input_node(0, &input_node)); in TEST()
158 TF_CHECK_OK(while_node->input_node(1, &input_node)); in TEST()
|
| D | compilability_check_util.cc | 277 const Node& while_node, FunctionLibraryRuntime* lib_runtime, in IsCompilableWhile() argument
284 while_node, "cond", "while_cond", encapsulating_function, lib_runtime, in IsCompilableWhile()
290 while_node, "body", "while_body", encapsulating_function, lib_runtime, in IsCompilableWhile()
|
| D | compilability_check_util.h | 217 bool IsCompilableWhile(const Node& while_node,
|
| D | extract_outside_compilation_pass.cc | 1477 Node* while_node) { in AddSendLoopPredToLoopCond() argument
1509 absl::StrCat("send_oc_while_cond_", while_node->name()), "XlaSendToHost"); in AddSendLoopPredToLoopCond()
1542 while_node->ClearAttr("cond"); in AddSendLoopPredToLoopCond()
1543 while_node->AddAttr("cond", *loop_cond_func); in AddSendLoopPredToLoopCond()
1736 Node* while_node = host_graph.AddNode(while_def, &s); in BuildHostGraphForWhileNode() local
1738 host_graph.AddEdge(key_placeholder, 0, while_node, 0); in BuildHostGraphForWhileNode()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | encapsulate_tpu_computations_pass.cc | 1763 Node* while_node, FunctionLibraryDefinition* fld) { in FindArgsToLiftForWhileNode() argument
1767 TF_RETURN_IF_ERROR(GetNodeAttr(while_node->def(), "T", &dtypes)); in FindArgsToLiftForWhileNode()
1777 TF_RETURN_IF_ERROR(GetNodeAttr(while_node->def(), "cond", &cond_func)); in FindArgsToLiftForWhileNode()
1781 " for while node ", while_node->DebugString()); in FindArgsToLiftForWhileNode()
1797 TF_RETURN_IF_ERROR(GetNodeAttr(while_node->def(), "body", &body_func)); in FindArgsToLiftForWhileNode()
1801 " for while node ", while_node->DebugString()); in FindArgsToLiftForWhileNode()
2043 void RemoveOutputIdentityNodesForWhileV2(Graph* g, Node* while_node) { in RemoveOutputIdentityNodesForWhileV2() argument
2045 for (const Edge* e : while_node->out_edges()) { in RemoveOutputIdentityNodesForWhileV2()
2056 g->AddControlEdge(while_node, out_edge->dst()); in RemoveOutputIdentityNodesForWhileV2()
2061 g->AddEdge(while_node, e->src_output(), dst, dst_input); in RemoveOutputIdentityNodesForWhileV2()
[all …]
|
| /external/tensorflow/tensorflow/python/autograph/pyct/static_analysis/ |
| D | activity_test.py | 265 while_node = node.body[1]
267 anno.getanno(while_node, NodeAnno.BODY_SCOPE), ('b',), ('b', 'c'))
269 anno.getanno(while_node, NodeAnno.BODY_SCOPE).parent, ('a', 'b', 'c'),
272 anno.getanno(while_node, NodeAnno.COND_SCOPE), ('b',), ())
|