Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dtuple_points_to_analysis_test.cc67 points_to_analysis_->GetPointsToSet(instruction).element(index); in GetBuffer()
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()
142 points_to_analysis_->GetPointsToSet(constant2).element({}), {constant2}); in TEST_F()
144 points_to_analysis_->GetPointsToSet(constant2).tuple_sources({}).empty()); in TEST_F()
146 EXPECT_EQ(3, points_to_analysis_->GetPointsToSet(tuple).size()); in TEST_F()
147 EXPECT_FALSE(points_to_analysis_->GetPointsToSet(tuple).IsAmbiguous()); in TEST_F()
[all …]
Dtuple_points_to_analysis.cc184 const PointsToSet& points_to_set = GetPointsToSet(instruction);
299 GetPointsToSet(recv_done->operand(0)); in HandleRecvDone()
336 const PointsToSet& operand_points_to_set = GetPointsToSet(send->operand(0)); in HandleSend()
428 const PointsToSet& TuplePointsToAnalysis::GetPointsToSet( in GetPointsToSet() function in xla::TuplePointsToAnalysis
446 const auto& buffers = GetPointsToSet(instruction).element(index); in InstructionDefinesBufferAtIndex()
482 const auto& buffers = GetPointsToSet(instruction).element(index); in GetBufferDefinedAt()
505 GetPointsToSet(instruction) in GatherBuffersDefinedByInstruction()
534 const PointsToSet& src_points_to_set = GetPointsToSet(src); in CreateCopiedPointsToSet()
579 const PointsToSet& points_to_set = GetPointsToSet(instruction); in InstructionToString()
639 GetPointsToSet(instruction).element(index); in GetAllUsesOfInstructionAtIndex()
Dbuffer_liveness.cc57 points_to_analysis_->GetPointsToSet(instruction) in Analyze()
68 points_to_analysis_->GetPointsToSet(root).CreateFlattenedSet(); in Analyze()
Dbuffer_assignment.cc70 const auto& points_to = points_to_analysis.GetPointsToSet(instruction); in AddBufferToColocatedSet()
336 const PointsToSet& BufferAssignment::GetPointsToSet( in GetPointsToSet() function in xla::BufferAssignment
338 return points_to_analysis().GetPointsToSet(instruction); in GetPointsToSet()
377 const PointsToSet& points_to_set = points_to_analysis().GetPointsToSet(hlo); in GetInstructionAllocation()
397 GetPointsToSet(instruction).element(index)) { in HasAllocationAt()
416 GetPointsToSet(instruction).element(index)) { in GetUniqueSlice()
771 assignment->liveness().points_to_analysis().GetPointsToSet( in MayInterfereAcrossSubcomputations()
779 assignment->liveness().points_to_analysis().GetPointsToSet(a_ancestor); in MayInterfereAcrossSubcomputations()
Dbuffer_assignment.h412 return GetPointsToSet(instruction).element(index); in GetSourceBuffers()
499 const PointsToSet& GetPointsToSet(const HloInstruction* instruction) const;
Dbuffer_liveness_test.cc43 .GetPointsToSet(instruction) in GetBuffer()
77 points_to_analysis.GetPointsToSet(a).element(index); in TupleElementsMayInterfere()
79 points_to_analysis.GetPointsToSet(b).element(index); in TupleElementsMayInterfere()
Dhlo_memory_scheduler.cc119 points_to_analysis.GetPointsToSet(operand).ForEachElement( in ListScheduler()
146 points_to_analysis.GetPointsToSet(computation->root_instruction()) in ListScheduler()
Dheap_simulator.cc142 points_to_analysis.GetPointsToSet(instruction); in RunComputation()
158 points_to_analysis.GetPointsToSet(user); in RunComputation()
169 points_to_analysis.GetPointsToSet(root).CreateFlattenedSet()); in RunComputation()
Dlayout_assignment.cc109 points_to_analysis_.GetPointsToSet(inst).ForEachElement( in LayoutConstraints()
133 const auto& points_to_set = points_to_analysis_.GetPointsToSet(instruction); in GetBufferSet()
309 points_to_analysis_.GetPointsToSet(instruction).element(index); in SetInstructionLayout()
891 points_to_analysis->GetPointsToSet(instruction); in CheckLayouts()
1200 constraints->points_to_analysis().GetPointsToSet(instruction); in PropagateUseConstraintToDefs()
1501 points_to_analysis.GetPointsToSet(instruction).element(index); in InferArrayLayout()
Dtuple_points_to_analysis.h191 const PointsToSet& GetPointsToSet(
Dhlo_rematerialization.cc515 points_to_analysis.GetPointsToSet(computation_->root_instruction()) in MemoryUsageTracker()
531 points_to_analysis.GetPointsToSet(instruction->operand(0)); in MemoryUsageTracker()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_executable.cc357 return assignment_->points_to_analysis().GetPointsToSet( in GetRootPointsToSet()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_executable.cc358 return assignment_->points_to_analysis().GetPointsToSet( in GetRootPointsToSet()