HomeSort by relevance Sort by last modified time
    Searched defs:call_graph (Results 1 - 9 of 9) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
defuser.cc 25 #include "tensorflow/compiler/xla/service/call_graph.h"
96 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module); local
97 TF_RETURN_IF_ERROR(call_graph->VisitNodes(
flatten_call_graph.cc 18 #include "tensorflow/compiler/xla/service/call_graph.h"
117 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module); local
118 TF_RETURN_IF_ERROR(call_graph->VisitNodes(FlattenNode));
hlo_dataflow_analysis.h 28 #include "tensorflow/compiler/xla/service/call_graph.h"
117 const CallGraph& call_graph() const { return *call_graph_; } function in class:xla::HloDataflowAnalysis
hlo_ordering.h 23 #include "tensorflow/compiler/xla/service/call_graph.h"
70 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"
138 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module); local
143 call_graph->VisitNodes([&](const CallGraphNode& node) -> Status {
call_graph.cc 16 #include "tensorflow/compiler/xla/service/call_graph.h"
239 auto call_graph = WrapUnique<CallGraph>(new CallGraph(module)); local
246 auto it_added = call_graph->node_indices_.insert(
247 {computation, call_graph->nodes_.size()});
251 call_graph->nodes_.emplace_back(computation);
255 call_graph->nodes_.back().AddCallSiteForInstruction(instruction);
262 call_graph->GetNode(computation).callsites()) {
265 call_graph->GetNode(callee).AddCallerCallSite(callsite);
270 call_graph->SetCallContexts();
271 XLA_VLOG_LINES(1, call_graph->ToString())
    [all...]
flatten_call_graph_test.cc 19 #include "tensorflow/compiler/xla/service/call_graph.h"
179 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module.get()); local
180 const CallGraphNode& cond_node = call_graph->GetNode(cond_computation);
187 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module.get()); local
188 const CallGraphNode& cond_node = call_graph->GetNode(cond_computation);
216 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module.get()); local
219 const CallGraphNode& c_node = call_graph->GetNode(c_computation);
222 const CallGraphNode& b_node = call_graph->GetNode(b_computation);
248 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module.get()); local
252 const CallGraphNode& sub_node = call_graph->GetNode(sub_computation)
    [all...]
call_graph_test.cc 16 #include "tensorflow/compiler/xla/service/call_graph.h"
99 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module.get()); local
100 EXPECT_EQ(1, call_graph->nodes().size());
101 EXPECT_TRUE(call_graph->IsFlattened());
103 const CallGraphNode& node = call_graph->GetNode(computation);
121 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module.get()); local
122 EXPECT_EQ(2, call_graph->nodes().size());
124 const CallGraphNode& entry_node = call_graph->GetNode(entry_computation);
129 call_graph->GetNode(unreachable_computation);
143 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module.get()) local
172 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module.get()); local
213 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module.get()); local
262 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module.get()); local
331 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module.get()); local
455 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module.get()); local
485 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module.get()); local
502 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module.get()); local
536 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module.get()); local
    [all...]
copy_insertion.cc 1064 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module); local
1157 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module); local
    [all...]

Completed in 73 milliseconds