Home
last modified time | relevance | path

Searched refs:dataflow_analysis (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_dataflow_analysis.cc843 auto dataflow_analysis = absl::WrapUnique(new HloDataflowAnalysis( in Run() local
846 TF_RETURN_IF_ERROR(dataflow_analysis->InitializeInstructionValueSets()); in Run()
847 dataflow_analysis->Propagate(); in Run()
850 dataflow_analysis->DeleteMarkedValues(); in Run()
857 dataflow_analysis->next_value_id_); in Run()
861 dataflow_analysis->GetInstructionValueSet(instruction)) { in Run()
873 for (auto& pair : dataflow_analysis->values_) { in Run()
880 dataflow_analysis->values_vector_.reserve(dataflow_analysis->values_.size()); in Run()
881 for (auto& pair : dataflow_analysis->values_) { in Run()
882 dataflow_analysis->values_vector_.push_back(&pair.second); in Run()
[all …]
Dhlo_alias_analysis.h85 const HloDataflowAnalysis& dataflow_analysis() const { in dataflow_analysis() function
Dcopy_insertion.cc262 if (!IndicesToCopyForWhile(alias_analysis.dataflow_analysis(), xla_while, in AddCopiesForWhile()
477 : dataflow_(alias_analysis.dataflow_analysis()), ordering_(ordering) { in CopyRemover()
1014 for (const HloValue* value : alias_analysis->dataflow_analysis().values()) { in AddSpecialCaseCopies()
1056 alias_analysis->dataflow_analysis().GetInstructionValueSet(root)) { in AddSpecialCaseCopies()
Dhlo_alias_analysis.cc502 BufferValueMap buffer_map(module, alias_analysis->dataflow_analysis()); in Run()
573 dataflow_analysis())) { in HasLiveRangeInterference()
Dhlo_alias_analysis_test.cc61 for (const HloValue* value : analysis_->dataflow_analysis()
86 return analysis_->dataflow_analysis().GetValueDefinedAt(instruction, index);
100 analysis_->dataflow_analysis())) { in AnyValuesInSameBufferInterfere()