Home
last modified time | relevance | path

Searched refs:GetPointsToSet (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dtuple_points_to_analysis_test.cc69 points_to_analysis_->GetPointsToSet(instruction).element(index); in GetBuffer()
135 EXPECT_EQ(1, points_to_analysis_->GetPointsToSet(constant1).size()); in TEST_F()
137 points_to_analysis_->GetPointsToSet(constant1).element({}), {constant1}); in TEST_F()
139 points_to_analysis_->GetPointsToSet(constant1).tuple_sources({}).empty()); in TEST_F()
140 EXPECT_TRUE(points_to_analysis_->GetPointsToSet(tuple).IsDistinct()); in TEST_F()
142 EXPECT_EQ(1, points_to_analysis_->GetPointsToSet(constant2).size()); in TEST_F()
144 points_to_analysis_->GetPointsToSet(constant2).element({}), {constant2}); in TEST_F()
146 points_to_analysis_->GetPointsToSet(constant2).tuple_sources({}).empty()); in TEST_F()
148 EXPECT_EQ(3, points_to_analysis_->GetPointsToSet(tuple).size()); in TEST_F()
149 EXPECT_FALSE(points_to_analysis_->GetPointsToSet(tuple).IsAmbiguous()); in TEST_F()
[all …]
Dtuple_points_to_analysis.cc187 const PointsToSet& points_to_set = GetPointsToSet(instruction);
302 GetPointsToSet(recv_done->operand(0)); in HandleRecvDone()
324 GetPointsToSet(copy_start->operand(0)); in HandleCopyStart()
348 GetPointsToSet(copy_done->operand(0)); in HandleCopyDone()
386 const PointsToSet& operand_points_to_set = GetPointsToSet(send->operand(0)); in HandleSend()
508 const PointsToSet& TuplePointsToAnalysis::GetPointsToSet( in GetPointsToSet() function in xla::TuplePointsToAnalysis
526 const auto& buffers = GetPointsToSet(instruction).element(index); in InstructionDefinesBufferAtIndex()
562 const auto& buffers = GetPointsToSet(instruction).element(index); in GetBufferDefinedAt()
585 GetPointsToSet(instruction) in GatherBuffersDefinedByInstruction()
614 const PointsToSet& src_points_to_set = GetPointsToSet(src); in CreateCopiedPointsToSet()
[all …]
Dtuple_points_to_analysis.h192 const PointsToSet& GetPointsToSet(
Dhlo_memory_scheduler.cc122 points_to_analysis.GetPointsToSet(operand).ForEachElement( in ListScheduler()
150 points_to_analysis.GetPointsToSet(computation->root_instruction()) in ListScheduler()
Dlayout_assignment.cc114 points_to_analysis_.GetPointsToSet(inst).ForEachElement( in LayoutConstraints()
138 const auto& points_to_set = points_to_analysis_.GetPointsToSet(instruction); in GetBufferSet()
319 points_to_analysis_.GetPointsToSet(instruction).element(index); in SetInstructionLayout()
944 points_to_analysis->GetPointsToSet(instruction); in CheckLayouts()
1258 constraints->points_to_analysis().GetPointsToSet(instruction); in PropagateUseConstraintToDefs()
1642 points_to_analysis.GetPointsToSet(instruction).element(index); in InferArrayLayout()
Dhlo_rematerialization.cc794 points_to_analysis.GetPointsToSet(computation_->root_instruction()) in MemoryUsageTracker()
809 points_to_analysis.GetPointsToSet(instruction->operand(0)); in MemoryUsageTracker()
852 points_to_analysis.GetPointsToSet(instruction).CreateFlattenedSet()) { in MemoryUsageTracker()