Home
last modified time | relevance | path

Searched refs:VisitNodes (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dcall_graph_test.cc375 TF_ASSERT_OK(call_graph->VisitNodes([&visited](const CallGraphNode& node) { in TEST_F()
509 TF_ASSERT_OK(call_graph->VisitNodes([&visited](const CallGraphNode& node) { in TEST_F()
528 TF_ASSERT_OK(call_graph->VisitNodes( in TEST_F()
541 TF_ASSERT_OK(call_graph->VisitNodes( in TEST_F()
559 Status status = call_graph->VisitNodes( in TEST_F()
Ddefuser.cc98 TF_RETURN_IF_ERROR(call_graph->VisitNodes( in Run()
Dconditional_to_select.cc70 call_graph->VisitNodes([&](const CallGraphNode& node) -> Status { in Run()
Dflatten_call_graph.cc121 TF_RETURN_IF_ERROR(call_graph->VisitNodes(FlattenNode)); in Run()
Dcall_inliner.cc142 call_graph->VisitNodes([&](const CallGraphNode& node) -> Status { in Run()
Dcall_graph.h200 Status VisitNodes(const VisitorFunction& visitor_func,
Dcall_graph.cc337 Status CallGraph::VisitNodes(const VisitorFunction& visitor_func, in VisitNodes() function in xla::CallGraph
Dbuffer_assignment.cc1277 TF_RETURN_IF_ERROR(call_graph->VisitNodes([&](const CallGraphNode& node) { in AssignBuffersForComputations()
Dhlo_rematerialization.cc2040 TF_RETURN_IF_ERROR(call_graph_->VisitNodes( in Run()