Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dtuple_points_to_analysis_test.cc61 points_to_analysis_ = TuplePointsToAnalysis::Run(module_.get()).value(); in RunAnalysis()
69 points_to_analysis_->GetPointsToSet(instruction).element(index); in GetBuffer()
111 points_to_analysis_->GetBufferDefinedAt(instruction, index) in ExpectHasBufferAliases()
118 EXPECT_THAT(points_to_analysis_->GetBufferAliases(*buffer), in ExpectHasBufferAliases()
123 std::unique_ptr<TuplePointsToAnalysis> points_to_analysis_; member in xla::__anond2b178470111::TuplePointsToAnalysisTest
Dlayout_assignment.cc146 const auto& points_to_set = points_to_analysis_->GetPointsToSet(instruction); in GetBufferSet()
185 TF_RETURN_IF_ERROR(points_to_analysis_->VerifyBuffer(buffer)); in SetBufferLayout()
380 points_to_analysis_->GetPointsToSet(instruction).element(index); in SetInstructionLayout()
418 points_to_analysis_->GetPointsToSet(instruction).element(index); in SetInstructionLayout()
478 points_to_analysis_->GetBuffersDefinedByInstruction(instruction)) { in ToString()
1487 points_to_analysis_->GetPointsToSet(instruction); in PropagateUseConstraintToDefs()
1564 points_to_analysis_->GetBufferDefinedAt(user, shape_index)); in PropagateOperandConstraint()
1658 if (!points_to_analysis_->InstructionDefinesBufferAtIndex( in PropagateOperandConstraint()
1666 points_to_analysis_->GetBufferDefinedAt(user, shape_index)); in PropagateOperandConstraint()
1687 if (!points_to_analysis_->InstructionDefinesBufferAtIndex( in PropagateOperandConstraint()
[all …]
Dhlo_rematerialization.h203 std::unique_ptr<TuplePointsToAnalysis> points_to_analysis_; variable
Dhlo_memory_scheduler.cc112 points_to_analysis_(points_to_analysis), in ListScheduler()
186 points_to_analysis_.GetBuffersDefinedByInstruction(instruction)) { in MakeReadyListEntry()
383 const TuplePointsToAnalysis& points_to_analysis_; member in xla::__anon454ff6160111::ListScheduler
Dlayout_assignment.h253 return *points_to_analysis_; in points_to_analysis()
614 std::unique_ptr<TuplePointsToAnalysis> points_to_analysis_; variable
Dhlo_rematerialization.cc1848 compact_shape_function_, *points_to_analysis_, in ComputePeakMemory()
1906 *points_to_analysis_, instruction_list, mode_); in RematerializeComputation()
2099 TF_ASSIGN_OR_RETURN(points_to_analysis_, TuplePointsToAnalysis::Run(module)); in Run()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment.cc195 points_to_analysis_->GetBufferDefinedAt(instr, /*index=*/{0})); in AddBackendConstraintsToDnnConvCustomCall()
349 points_to_analysis_->GetBufferDefinedAt(instruction, {})); in AddBackendConstraints()
353 points_to_analysis_->GetBufferDefinedAt(instruction, {i})); in AddBackendConstraints()