Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_dataflow_analysis.h105 const HloValueSet& GetValueSet(const HloInstruction* instruction,
107 const HloValueSet& GetValueSet(const HloPosition& position) const;
108 HloValueSet& GetValueSet(const HloPosition& position);
109 HloValueSet& GetValueSet(const HloInstruction* instruction,
116 return GetValueSet(instruction, index).GetUniqueValue();
120 return GetValue(GetValueSet(instruction, index).GetUniqueValue().id());
Dhlo_dataflow_analysis.cc80 const HloValueSet& value_set = GetValueSet(instruction, index); in ValueIsDefinedAt()
167 GetValueSet(instruction, /*index=*/{}); in ToString()
287 const HloValueSet& HloDataflowAnalysis::GetValueSet( in GetValueSet() function in xla::HloDataflowAnalysis
292 HloValueSet& HloDataflowAnalysis::GetValueSet(const HloInstruction* instruction, in GetValueSet() function in xla::HloDataflowAnalysis
297 const HloValueSet& HloDataflowAnalysis::GetValueSet( in GetValueSet() function in xla::HloDataflowAnalysis
299 return GetValueSet(position.instruction, position.index); in GetValueSet()
302 HloValueSet& HloDataflowAnalysis::GetValueSet(const HloPosition& position) { in GetValueSet() function in xla::HloDataflowAnalysis
303 return GetValueSet(position.instruction, position.index); in GetValueSet()
331 HloValueSet& value_set = GetValueSet(send, index); in UpdateSendValueSet()
353 GetValueSet(recv_done->operand(0), index); in UpdateRecvDoneValueSet()
[all …]
Dbfloat16_propagation.cc91 dataflow_->GetValueSet(root, entry.second).values()) { in RevertIfFusionInternalBF16Changes()
112 dataflow_->GetValueSet(inst, index).values()) { in RevertIfFusionInternalBF16Changes()
214 auto& value_set = dataflow_->GetValueSet(&hlo, index); in AllUsersConsumeBF16()
346 for (const auto* value : dataflow_->GetValueSet(hlo, index).values()) { in DetermineInstructionPrecision()
472 for (const auto* value : dataflow_->GetValueSet(root, index).values()) { in AdjustCalledComputationRoot()
518 for (const auto* value : dataflow_->GetValueSet(hlo, index).values()) { in ResolveInconsistencyOfAliasingBuffersHelper()
534 for (const auto* value : dataflow_->GetValueSet(hlo, index).values()) { in ResolveInconsistencyOfAliasingBuffersHelper()
Dcopy_insertion.cc179 if (dataflow.GetValueSet(init, index).values().size() > 1 || in IndicesToCopyForWhile()
180 dataflow.GetValueSet(xla_while, index).values().size() > 1) { in IndicesToCopyForWhile()
571 dataflow_.GetValueSet(instruction->operand(0)); in CreateCopyMap()
Dhlo_alias_analysis.cc364 dataflow_analysis_->GetValueSet(instruction, index).values()) { in ComputeBuffersAt()
Dhlo_alias_analysis_test.cc62 .GetValueSet(instruction, index)
Dhlo_dataflow_analysis_test.cc65 analysis_->GetValueSet(instruction, index).values()) {