Home	Sort by relevance Sort by last modified time

Full Search Definition Symbol File Path History   | | Help

    Searched defs:call_graph (Results 1 - 10 of 10) sorted by null

/external/tensorflow/tensorflow/compiler/xla/service/
defuser.cc	26 #include "tensorflow/compiler/xla/service/call_graph.h" 97 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module); local 98 TF_RETURN_IF_ERROR(call_graph->VisitNodes(
flatten_call_graph.cc	18 #include "tensorflow/compiler/xla/service/call_graph.h" 120 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module); local 121 TF_RETURN_IF_ERROR(call_graph->VisitNodes(FlattenNode));
hlo_ordering.h	24 #include "tensorflow/compiler/xla/service/call_graph.h" 72 const CallGraph& call_graph() const { return *call_graph_; } function in class:xla::HloOrdering
call_inliner.cc	20 #include "tensorflow/compiler/xla/service/call_graph.h" 139 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module); local 144 call_graph->VisitNodes([&](const CallGraphNode& node) -> Status {
hlo_dataflow_analysis.h	29 #include "tensorflow/compiler/xla/service/call_graph.h" 134 const CallGraph& call_graph() const { return *call_graph_; } function in class:xla::HloDataflowAnalysis
flatten_call_graph_test.cc	19 #include "tensorflow/compiler/xla/service/call_graph.h" 180 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module.get()); local 181 const CallGraphNode& cond_node = call_graph->GetNode(cond_computation); 188 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module.get()); local 189 const CallGraphNode& cond_node = call_graph->GetNode(cond_computation); 217 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module.get()); local 220 const CallGraphNode& c_node = call_graph->GetNode(c_computation); 223 const CallGraphNode& b_node = call_graph->GetNode(b_computation); 249 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module.get()); local 253 const CallGraphNode& sub_node = call_graph->GetNode(sub_computation)     [all...]
call_graph.cc	16 #include "tensorflow/compiler/xla/service/call_graph.h" 278 auto call_graph = absl::WrapUnique<CallGraph>(new CallGraph(module)); local 285 auto it_added = call_graph->node_indices_.insert( 286 {computation, call_graph->nodes_.size()}); 290 call_graph->nodes_.emplace_back(computation); 294 call_graph->nodes_.back().AddCallSiteForInstruction(instruction); 301 call_graph->GetNode(computation).callsites()) { 304 call_graph->GetNode(callee).AddCallerCallSite(callsite); 309 call_graph->SetCallContexts(); 310 call_graph->SetNodeDepths()     [all...]
call_graph_test.cc	16 #include "tensorflow/compiler/xla/service/call_graph.h" 100 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module.get()); local 101 EXPECT_EQ(1, call_graph->nodes().size()); 102 EXPECT_TRUE(call_graph->IsFlattened()); 104 const CallGraphNode& node = call_graph->GetNode(computation); 123 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module.get()); local 124 EXPECT_EQ(2, call_graph->nodes().size()); 126 const CallGraphNode& entry_node = call_graph->GetNode(entry_computation); 132 call_graph->GetNode(unreachable_computation); 147 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module.get()) local 178 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module.get()); local 219 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module.get()); local 269 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module.get()); local 341 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module.get()); local 476 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module.get()); local 506 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module.get()); local 523 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module.get()); local 557 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module.get()); local     [all...]
copy_insertion.cc	985 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module); local 1113 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module); local 1152 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module); local 1240 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module); local     [all...]
buffer_assignment.cc	756 const CallGraph& call_graph = local 757 assignment->liveness().hlo_ordering().call_graph(); 761 call_graph.NearestAncestorsInSameComputation(a_buffer.instruction(), 768 call_graph.InstructionIsNestedIn(a_buffer.instruction(),     [all...]

Completed in 4766 milliseconds