Searched refs:CreateFlattenedSet (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | tuple_points_to_analysis_test.cc | 152 points_to_analysis_->GetPointsToSet(tuple).CreateFlattenedSet(), in TEST_F()
202 points_to_analysis_->GetPointsToSet(inner_tuple).CreateFlattenedSet(), in TEST_F()
214 points_to_analysis_->GetPointsToSet(tuple).CreateFlattenedSet(), in TEST_F()
260 ExpectHasTopLevelBuffers(points_to_set.CreateFlattenedSet(), in TEST_F()
281 ExpectHasTopLevelBuffers(points_to_set.CreateFlattenedSet(), {constant}); in TEST_F()
300 points_to_analysis_->GetPointsToSet(tuple).CreateFlattenedSet(), in TEST_F()
322 points_to_analysis_->GetPointsToSet(tuple).CreateFlattenedSet(), in TEST_F()
327 points_to_analysis_->GetPointsToSet(copy).CreateFlattenedSet(), in TEST_F()
353 points_to_analysis_->GetPointsToSet(send_done).CreateFlattenedSet(), in TEST_F()
405 ExpectHasTopLevelBuffers(points_to_set.CreateFlattenedSet(), in TEST_F()
[all …]
|
| D | buffer_liveness.cc | 58 .CreateFlattenedSet()) { in Analyze()
68 points_to_analysis_->GetPointsToSet(root).CreateFlattenedSet(); in Analyze()
|
| D | heap_simulator.cc | 143 const PointsToSet::BufferSet& buffer_set = points_to.CreateFlattenedSet(); in RunComputation()
159 for (const BufferValue* buffer : gte_points_to.CreateFlattenedSet()) { in RunComputation()
169 points_to_analysis.GetPointsToSet(root).CreateFlattenedSet()); in RunComputation()
|
| D | tuple_points_to_analysis.h | 73 BufferSet CreateFlattenedSet() const;
|
| D | tuple_points_to_analysis.cc | 74 return CreateFlattenedSet().size(); in size()
77 PointsToSet::BufferSet PointsToSet::CreateFlattenedSet() const { in CreateFlattenedSet() function in xla::PointsToSet
|
| D | hlo_memory_scheduler.cc | 147 .CreateFlattenedSet()) { in ListScheduler()
|
| D | hlo_rematerialization.cc | 516 .CreateFlattenedSet(); in MemoryUsageTracker()
|