Searched refs:CreateFlattenedSet (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | tuple_points_to_analysis_test.cc | 154 points_to_analysis_->GetPointsToSet(tuple).CreateFlattenedSet(), in TEST_F()
204 points_to_analysis_->GetPointsToSet(inner_tuple).CreateFlattenedSet(), in TEST_F()
216 points_to_analysis_->GetPointsToSet(tuple).CreateFlattenedSet(), in TEST_F()
262 ExpectHasTopLevelBuffers(points_to_set.CreateFlattenedSet(), in TEST_F()
283 ExpectHasTopLevelBuffers(points_to_set.CreateFlattenedSet(), {constant}); in TEST_F()
302 points_to_analysis_->GetPointsToSet(tuple).CreateFlattenedSet(), in TEST_F()
324 points_to_analysis_->GetPointsToSet(tuple).CreateFlattenedSet(), in TEST_F()
329 points_to_analysis_->GetPointsToSet(copy).CreateFlattenedSet(), in TEST_F()
381 points_to_analysis_->GetPointsToSet(send_done).CreateFlattenedSet(), in TEST_F()
433 ExpectHasTopLevelBuffers(points_to_set.CreateFlattenedSet(), in TEST_F()
[all …]
|
| D | tuple_points_to_analysis.h | 74 BufferSet CreateFlattenedSet() const;
|
| D | tuple_points_to_analysis.cc | 77 return CreateFlattenedSet().size(); in size()
80 PointsToSet::BufferSet PointsToSet::CreateFlattenedSet() const { in CreateFlattenedSet() function in xla::PointsToSet
|
| D | hlo_memory_scheduler.cc | 151 .CreateFlattenedSet()) { in ListScheduler()
|
| D | hlo_rematerialization.cc | 795 .CreateFlattenedSet(); in MemoryUsageTracker()
852 points_to_analysis.GetPointsToSet(instruction).CreateFlattenedSet()) { in MemoryUsageTracker()
|