Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dtuple_points_to_analysis_test.cc58 points_to_analysis_ = in RunAnalysis()
67 points_to_analysis_->GetPointsToSet(instruction).element(index); in GetBuffer()
109 points_to_analysis_->GetBufferDefinedAt(instruction, index) in ExpectHasBufferAliases()
115 EXPECT_THAT(points_to_analysis_->GetBufferAliases(*buffer), in ExpectHasBufferAliases()
120 std::unique_ptr<TuplePointsToAnalysis> points_to_analysis_; member in xla::__anon55903b910111::TuplePointsToAnalysisTest
133 EXPECT_EQ(1, points_to_analysis_->GetPointsToSet(constant1).size()); in TEST_F()
135 points_to_analysis_->GetPointsToSet(constant1).element({}), {constant1}); in TEST_F()
137 points_to_analysis_->GetPointsToSet(constant1).tuple_sources({}).empty()); in TEST_F()
138 EXPECT_TRUE(points_to_analysis_->GetPointsToSet(tuple).IsDistinct()); in TEST_F()
140 EXPECT_EQ(1, points_to_analysis_->GetPointsToSet(constant2).size()); in TEST_F()
[all …]
Dbuffer_liveness.cc47 TF_ASSIGN_OR_RETURN(points_to_analysis_, TuplePointsToAnalysis::Run(module_)); in Analyze()
57 points_to_analysis_->GetPointsToSet(instruction) in Analyze()
68 points_to_analysis_->GetPointsToSet(root).CreateFlattenedSet(); in Analyze()
95 TF_DCHECK_OK(points_to_analysis_->VerifyBuffer(a)); in live_range_strictly_before()
96 TF_DCHECK_OK(points_to_analysis_->VerifyBuffer(b)); in live_range_strictly_before()
102 for (const BufferAlias& alias : points_to_analysis_->GetBufferAliases(a)) { in live_range_strictly_before()
130 for (const BufferAlias& alias : points_to_analysis_->GetBufferAliases(a)) { in live_range_strictly_before()
169 TF_CHECK_OK(points_to_analysis_->VerifyBuffer(buffer)); in MaybeLiveOut()
Dbuffer_liveness.h62 return *points_to_analysis_; in points_to_analysis()
109 std::unique_ptr<TuplePointsToAnalysis> points_to_analysis_; variable
Dhlo_rematerialization.h121 std::unique_ptr<TuplePointsToAnalysis> points_to_analysis_; variable
Dhlo_memory_scheduler.cc109 points_to_analysis_(points_to_analysis), in ListScheduler()
182 points_to_analysis_.GetBuffersDefinedByInstruction(instruction)) { in MakeReadyListEntry()
371 const TuplePointsToAnalysis& points_to_analysis_; member in xla::__anon74ea0e200111::ListScheduler
Dlayout_assignment.h139 return points_to_analysis_; in points_to_analysis()
226 const TuplePointsToAnalysis& points_to_analysis_; variable
Dheap_simulator_test.cc266 points_to_analysis_ = in HeapSimulatorTracker()
278 *points_to_analysis_, zero_size) in HeapSimulatorTracker()
291 points_to_analysis_ = in RunWholeModule()
314 *points_to_analysis_, size_fn) in RunWholeModule()
323 return points_to_analysis_->GetBufferDefinedAt(instruction, index) in BufferAt()
350 std::unique_ptr<TuplePointsToAnalysis> points_to_analysis_; member in xla::__anon72fb25430111::HeapSimulatorTracker
Dlayout_assignment.cc106 : points_to_analysis_(points_to_analysis), computation_(computation) { in LayoutConstraints()
109 points_to_analysis_.GetPointsToSet(inst).ForEachElement( in LayoutConstraints()
133 const auto& points_to_set = points_to_analysis_.GetPointsToSet(instruction); in GetBufferSet()
160 TF_RETURN_IF_ERROR(points_to_analysis_.VerifyBuffer(buffer)); in SetBufferLayout()
309 points_to_analysis_.GetPointsToSet(instruction).element(index); in SetInstructionLayout()
367 points_to_analysis_.GetBuffersDefinedByInstruction(instruction)) { in ToString()
Dhlo_rematerialization.cc938 MemoryUsageTracker tracker(computation, size_function_, *points_to_analysis_, in ComputePeakMemory()
982 *points_to_analysis_, instruction_list); in RematerializeComputation()
1215 TF_ASSIGN_OR_RETURN(points_to_analysis_, TuplePointsToAnalysis::Run(module)); in Run()