| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | call_graph_test.cc | 100 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module.get()); in TEST_F() local
101 EXPECT_EQ(1, call_graph->nodes().size()); in TEST_F()
102 EXPECT_TRUE(call_graph->IsFlattened()); in TEST_F()
104 const CallGraphNode& node = call_graph->GetNode(computation); in TEST_F()
123 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module.get()); in TEST_F() local
124 EXPECT_EQ(2, call_graph->nodes().size()); in TEST_F()
126 const CallGraphNode& entry_node = call_graph->GetNode(entry_computation); in TEST_F()
132 call_graph->GetNode(unreachable_computation); in TEST_F()
147 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module.get()); in TEST_F() local
148 EXPECT_EQ(2, call_graph->nodes().size()); in TEST_F()
[all …]
|
| D | call_graph.cc | 278 auto call_graph = absl::WrapUnique<CallGraph>(new CallGraph(module)); in Build() local
285 auto it_added = call_graph->node_indices_.insert( in Build()
286 {computation, call_graph->nodes_.size()}); in Build()
290 call_graph->nodes_.emplace_back(computation); in Build()
294 call_graph->nodes_.back().AddCallSiteForInstruction(instruction); in Build()
301 call_graph->GetNode(computation).callsites()) { in Build()
304 call_graph->GetNode(callee).AddCallerCallSite(callsite); in Build()
309 call_graph->SetCallContexts(); in Build()
310 call_graph->SetNodeDepths(); in Build()
312 XLA_VLOG_LINES(1, call_graph->ToString()); in Build()
[all …]
|
| D | flatten_call_graph_test.cc | 180 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module.get()); in TEST_F() local
181 const CallGraphNode& cond_node = call_graph->GetNode(cond_computation); in TEST_F()
188 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module.get()); in TEST_F() local
189 const CallGraphNode& cond_node = call_graph->GetNode(cond_computation); in TEST_F()
217 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module.get()); in TEST_F() local
220 const CallGraphNode& c_node = call_graph->GetNode(c_computation); in TEST_F()
223 const CallGraphNode& b_node = call_graph->GetNode(b_computation); in TEST_F()
249 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module.get()); in TEST_F() local
253 const CallGraphNode& sub_node = call_graph->GetNode(sub_computation); in TEST_F()
|
| D | copy_insertion.cc | 985 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module); in AddSpecialCaseCopies() local
986 return AddSpecialCaseCopies(*call_graph, module); in AddSpecialCaseCopies()
989 Status CopyInsertion::AddSpecialCaseCopies(const CallGraph& call_graph, in AddSpecialCaseCopies() argument
1026 const CallGraphNode& node = call_graph.GetNode(computation); in AddSpecialCaseCopies()
1113 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module); in RemoveUnnecessaryCopies() local
1152 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module); in Run() local
1153 if (!call_graph->IsFlattened()) { in Run()
1190 TF_RETURN_IF_ERROR(AddSpecialCaseCopies(*call_graph, module)); in Run()
1218 const CallGraph& call_graph) { in IsWhileBody() argument
1219 const CallGraphNode& node = call_graph.GetNode(computation); in IsWhileBody()
[all …]
|
| D | hlo_liveness_analysis.cc | 197 Workset* workset, CallGraph* call_graph) { in PropagateLivenessToParameterCallers() argument
200 call_graph->GetNode(instruction->parent()); in PropagateLivenessToParameterCallers()
228 Workset* workset, CallGraph* call_graph) { in PropagateLivenessThroughControlFlow() argument
230 call_graph->GetNode(instruction->parent()); in PropagateLivenessThroughControlFlow()
|
| D | defuser.cc | 97 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module); in Run() local
98 TF_RETURN_IF_ERROR(call_graph->VisitNodes( in Run()
|
| D | flatten_call_graph.cc | 120 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module); in Run() local
121 TF_RETURN_IF_ERROR(call_graph->VisitNodes(FlattenNode)); in Run()
|
| D | call_inliner.cc | 139 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module); in Run() local
144 call_graph->VisitNodes([&](const CallGraphNode& node) -> Status { in Run()
|
| D | copy_insertion.h | 99 virtual Status AddSpecialCaseCopies(const CallGraph& call_graph,
|
| D | hlo_alias_analysis.cc | 230 dataflow_.call_graph().GetNode(computation); in ComputeWhileAliasedBuffers()
249 dataflow_.call_graph().GetNode(computation); in ComputeWhileAliasedBuffers()
284 dataflow_.call_graph().GetNode(computation); in ComputeConditionalAliasedBuffers()
|
| D | hlo_ordering.h | 72 const CallGraph& call_graph() const { return *call_graph_; } in call_graph() function
|
| D | hlo_dataflow_analysis.h | 134 const CallGraph& call_graph() const { return *call_graph_; } in call_graph() function
|
| D | BUILD | 564 name = "call_graph",
565 srcs = ["call_graph.cc"],
566 hdrs = ["call_graph.h"],
584 ":call_graph",
604 ":call_graph",
618 ":call_graph",
653 ":call_graph",
1150 ":call_graph",
1179 ":call_graph",
1934 ":call_graph",
[all …]
|
| D | buffer_assignment.cc | 756 const CallGraph& call_graph = in MayInterfereAcrossSubcomputations() local
757 assignment->liveness().hlo_ordering().call_graph(); in MayInterfereAcrossSubcomputations()
761 call_graph.NearestAncestorsInSameComputation(a_buffer.instruction(), in MayInterfereAcrossSubcomputations()
768 call_graph.InstructionIsNestedIn(a_buffer.instruction(), in MayInterfereAcrossSubcomputations()
|
| D | buffer_assignment_test.cc | 2316 std::unique_ptr<CallGraph> call_graph = CallGraph::Build(module.get()); in TEST_F() local
|
| /external/google-fruit/extras/scripts/ |
| D | analyze_template_instantiations_clang_diagnostics.py | 362 call_graph = {}
371 if called in call_graph and call_graph[called] != caller:
375 call_graph[called] = caller
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | BUILD | 648 "//tensorflow/compiler/xla/service:call_graph",
|