| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_dataflow_analysis.h | 113 const HloValueSet& GetValueSet(const HloInstruction* instruction,
115 const HloValueSet& GetValueSet(const HloPosition& position) const;
116 HloValueSet& GetValueSet(const HloPosition& position);
117 HloValueSet& GetValueSet(const HloInstruction* instruction,
124 return GetValueSet(instruction, index).GetUniqueValue();
128 return GetValue(GetValueSet(instruction, index).GetUniqueValue().id());
|
| D | hlo_dataflow_analysis.cc | 125 const HloValueSet& value_set = GetValueSet(instruction, index); in ValueIsDefinedAt()
213 GetValueSet(instruction, /*index=*/{}); in ToString()
368 const HloValueSet& HloDataflowAnalysis::GetValueSet( in GetValueSet() function in xla::HloDataflowAnalysis
373 HloValueSet& HloDataflowAnalysis::GetValueSet(const HloInstruction* instruction, in GetValueSet() function in xla::HloDataflowAnalysis
378 const HloValueSet& HloDataflowAnalysis::GetValueSet( in GetValueSet() function in xla::HloDataflowAnalysis
380 return GetValueSet(position.instruction, position.index); in GetValueSet()
383 HloValueSet& HloDataflowAnalysis::GetValueSet(const HloPosition& position) { in GetValueSet() function in xla::HloDataflowAnalysis
384 return GetValueSet(position.instruction, position.index); in GetValueSet()
426 HloValueSet& value_set = GetValueSet(send, index); in UpdateSendValueSet()
441 const HloValueSet& operand_value_set = GetValueSet( in UpdateCustomCallValueSet()
[all …]
|
| D | bfloat16_propagation.cc | 93 dataflow_->GetValueSet(root, entry.second).values()) { in RevertIfFusionInternalBF16Changes()
114 dataflow_->GetValueSet(inst, index).values()) { in RevertIfFusionInternalBF16Changes()
243 auto& value_set = dataflow_->GetValueSet(&hlo, index); in AllUsersConsumeBF16()
415 for (const auto* value : dataflow_->GetValueSet(hlo, index).values()) { in DetermineInstructionPrecision()
541 for (const auto* value : dataflow_->GetValueSet(root, index).values()) { in AdjustCalledComputationRoot()
595 for (const auto* value : dataflow_->GetValueSet(hlo, index).values()) { in ResolveInconsistencyOfAliasingBuffersHelper()
617 ->GetValueSet(hlo->operand(operand.operand_number), in ResolveInconsistencyOfAliasingBuffersHelper()
642 dataflow_->GetValueSet(hlo, index).values()) { in ResolveInconsistencyOfAliasingBuffersHelper()
|
| D | loop_schedule_linearizer.cc | 92 if (dataflow.GetValueSet(root, index).values().size() > 1 || in AddControlEdgesForLoopWrites()
93 dataflow.GetValueSet(input, index).values().size() > 1) { in AddControlEdgesForLoopWrites()
|
| D | copy_insertion.cc | 179 if (dataflow.GetValueSet(init, index).values().size() > 1 || in IndicesToCopyForWhile()
180 dataflow.GetValueSet(xla_while, index).values().size() > 1) { in IndicesToCopyForWhile()
207 CHECK_EQ(dataflow.GetValueSet(xla_conditional, index).values().size(), 1); in IndicesToCopyForConditional()
209 auto value = dataflow.GetValueSet(xla_conditional, index).values()[0]; in IndicesToCopyForConditional()
589 dataflow_.GetValueSet(instruction->operand(0)); in CreateCopyMap()
|
| 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()
|
| D | buffer_assignment.h | 432 return dataflow_analysis().GetValueSet(instruction, index).values(); in GetSourceBuffers()
|
| D | hlo_alias_analysis.cc | 399 dataflow_analysis_->GetValueSet(instruction, index).values()) { in ComputeBuffersAt()
|
| D | heap_simulator.cc | 252 dataflow_analysis.GetValueSet(operand); in RunComputation()
|
| D | hlo_alias_analysis_test.cc | 62 .GetValueSet(instruction, index)
|
| D | hlo_dataflow_analysis_test.cc | 68 analysis_->GetValueSet(instruction, index).values()) {
|