| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_dataflow_analysis.cc | 1112 auto dataflow_analysis = absl::WrapUnique(new HloDataflowAnalysis( in Run() local
1115 TF_RETURN_IF_ERROR(dataflow_analysis->InitializeInstructionValueSets()); in Run()
1116 dataflow_analysis->Propagate(); in Run()
1117 dataflow_analysis->OptimizePhiValues(); in Run()
1120 dataflow_analysis->DeleteMarkedValues(); in Run()
1127 dataflow_analysis->next_value_id_); in Run()
1131 dataflow_analysis->GetInstructionValueSet(instruction)) { in Run()
1143 for (auto& pair : dataflow_analysis->values_) { in Run()
1150 dataflow_analysis->values_vector_.reserve(dataflow_analysis->values_.size()); in Run()
1151 for (auto& pair : dataflow_analysis->values_) { in Run()
[all …]
|
| D | memory_space_propagation.cc | 22 TF_ASSIGN_OR_RETURN(auto dataflow_analysis, in Run()
24 dataflow_analysis_ = std::move(dataflow_analysis); in Run()
|
| D | buffer_assignment.h | 432 return dataflow_analysis().GetValueSet(instruction, index).values(); in GetSourceBuffers()
457 const HloDataflowAnalysis& dataflow_analysis() const { in dataflow_analysis() function
458 return alias_analysis_->dataflow_analysis(); in dataflow_analysis()
591 for (HloValue* value : alias_analysis->dataflow_analysis().values()) { in DefaultColorer()
|
| D | hlo_live_range.cc | 131 for (const HloValue* value : alias_analysis_.dataflow_analysis().values()) { in CalculateBufferStartEndMap()
222 for (const HloValue* value : alias_analysis_.dataflow_analysis().values()) { in ToString()
|
| D | heap_simulator.cc | 164 HloDataflowAnalysis& dataflow_analysis = alias_analysis.dataflow_analysis(); in RunComputation() local
185 values_to_assign.reserve(dataflow_analysis.values().size()); in RunComputation()
187 for (const HloValue* value : dataflow_analysis.values()) { in RunComputation()
252 dataflow_analysis.GetValueSet(operand); in RunComputation()
290 dataflow_analysis.CanShareOperandBufferWithUser( in RunComputation()
|
| D | buffer_assignment.cc | 422 dataflow_analysis().GetValueSet(instruction, index).values()) { in GetAllSlices()
440 dataflow_analysis().GetValueSet(hlo, shape_index).values()[0]; in GetInstructionAllocation()
459 dataflow_analysis().GetValueSet(instruction, index).values()) { in HasAllocationAt()
478 dataflow_analysis().GetValueSet(instruction, index).values()) { in GetUniqueSlice()
875 const HloDataflowAnalysis& dataflow = this->dataflow_analysis(); in ToProto()
942 assignment->dataflow_analysis().CanShareOperandBufferWithUser( in LiveRangeInterferes()
1060 assignment->dataflow_analysis())) { in MaybeAssignBuffer()
1651 XLA_VLOG_LINES(3, alias_analysis->dataflow_analysis().ToString()); in CreateAssignment()
1666 assignment->alias_analysis().dataflow_analysis().ToString()); in CreateAssignment()
|
| D | hlo_alias_analysis.h | 84 HloDataflowAnalysis& dataflow_analysis() const { return *dataflow_analysis_; } in dataflow_analysis() function
|
| D | copy_insertion.cc | 285 if (!IndicesToCopyForWhile(alias_analysis.dataflow_analysis(), xla_while, in AddCopiesForWhile()
489 : dataflow_(alias_analysis.dataflow_analysis()), ordering_(ordering) { in CopyRemover()
989 if (!IndicesToCopyForConditional(alias_analysis.dataflow_analysis(), in AddCopiesForConditional()
1072 for (const HloValue* value : alias_analysis->dataflow_analysis().values()) { in AddSpecialCaseCopies()
1130 alias_analysis->dataflow_analysis().GetInstructionValueSet(root)) { in AddSpecialCaseCopies()
|
| D | loop_schedule_linearizer.cc | 72 HloDataflowAnalysis& dataflow = alias_analysis.dataflow_analysis(); in AddControlEdgesForLoopWrites()
|
| D | hlo_alias_analysis.cc | 537 BufferValueMap buffer_map(module, alias_analysis->dataflow_analysis()); in Run()
673 dataflow_analysis())) { in HasLiveRangeInterference()
|
| D | memory_space_assignment.cc | 900 &alias_analysis_.dataflow_analysis().GetUniqueValueAt( in IsUseAllowedInAlternateMemory()
933 &alias_analysis_.dataflow_analysis().GetUniqueValueAt( in IsUseAllowedInAlternateMemory()
959 &alias_analysis_.dataflow_analysis().GetUniqueValueAt( in IsUseAllowedInAlternateMemory()
1657 &alias_analysis_.dataflow_analysis().GetUniqueValueAt( in AliasedRequiredAssignmentForUse()
1714 &alias_analysis_.dataflow_analysis().GetUniqueValueAt(instruction, index); in AddRequiredAssignment()
1724 const HloModule& module = alias_analysis_.dataflow_analysis().module(); in AddInputAndOutputRequiredAssignments()
1785 const HloModule& module = alias_analysis_.dataflow_analysis().module(); in AreIntervalsReservedInAlternateMemory()
2601 TF_ASSIGN_OR_RETURN(std::unique_ptr<HloDataflowAnalysis> dataflow_analysis, in CalculateAsyncCopyStats()
2611 options_.size_fn(dataflow_analysis->GetUniqueValueAt(instruction)); in CalculateAsyncCopyStats()
|
| D | hlo_alias_analysis_test.cc | 61 for (const HloValue* value : analysis_->dataflow_analysis()
86 return analysis_->dataflow_analysis().GetValueDefinedAt(instruction, index);
100 analysis_->dataflow_analysis())) { in AnyValuesInSameBufferInterfere()
|
| D | hlo_live_range_test.cc | 59 return &alias_analysis_->dataflow_analysis().GetUniqueValueAt(instruction, in BufferAt()
|
| D | buffer_assignment_test.cc | 588 id < alias_analysis->dataflow_analysis().values().size(); id++) { in TEST_F()
589 auto& value = alias_analysis->dataflow_analysis().GetValue(id); in TEST_F()
653 id < alias_analysis->dataflow_analysis().values().size(); id++) { in TEST_F()
654 auto& value = alias_analysis->dataflow_analysis().GetValue(id); in TEST_F()
|
| D | memory_space_assignment_test.cc | 3855 for (const HloValue* value : alias_analysis->dataflow_analysis().values()) { in TEST_P()
5140 std::unique_ptr<HloDataflowAnalysis> dataflow_analysis, in TEST_P()
5143 dataflow_analysis->GetValueDefinedAt( in TEST_P()
5200 std::unique_ptr<HloDataflowAnalysis> dataflow_analysis, in TEST_P()
5203 dataflow_analysis->GetValueDefinedAt( in TEST_P()
|
| D | hlo_dataflow_analysis_test.cc | 1699 auto& dataflow_analysis = RunAnalysis(GetParam()); in TEST_P() local
1700 auto set = dataflow_analysis.GetFlattenedValueSet( in TEST_P()
|
| D | heap_simulator_test.cc | 335 return &alias_analysis_->dataflow_analysis().GetUniqueValueAt(instruction, in BufferAt()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_executable.cc | 409 return assignment_->dataflow_analysis().GetInstructionValueSet( in GetRootValueSet()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_executable.cc | 589 assignment.dataflow_analysis().GetInstructionValueSet(root); in GetOutputInfo()
|