Home
last modified time | relevance | path

Searched refs:dataflow_ (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_alias_analysis.cc64 : module_(module), dataflow_(dataflow) { in BufferValueMap()
65 buffers_.reserve(dataflow_.values().size()); in BufferValueMap()
66 value_to_buffer_number_.reserve(dataflow_.values().size()); in BufferValueMap()
67 for (const HloValue* value : dataflow_.values()) { in BufferValueMap()
77 for (const HloValue* value : dataflow_.values()) { in MergeAliasedBuffers()
181 return dataflow_.GetUniqueValueAt( in ComputeInputOutputAliasedBuffers()
218 dataflow_.GetUniqueValueAt(use.instruction, use.operand_index); in ComputeWhileAliasedBuffers()
230 dataflow_.call_graph().GetNode(computation); in ComputeWhileAliasedBuffers()
236 const HloValue& while_value = dataflow_.GetUniqueValueAt( in ComputeWhileAliasedBuffers()
249 dataflow_.call_graph().GetNode(computation); in ComputeWhileAliasedBuffers()
[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()
754 TF_ASSIGN_OR_RETURN(dataflow_, HloDataflowAnalysis::Run(*module)); in Run()
Dcopy_insertion.cc477 : dataflow_(alias_analysis.dataflow_analysis()), ordering_(ordering) { in CopyRemover()
496 dataflow_) || in CopyRemover()
498 dataflow_)) in CopyRemover()
571 dataflow_.GetValueSet(instruction->operand(0)); in CreateCopyMap()
575 value_to_node.at(&dataflow_.GetUniqueValueAt(instruction)); in CreateCopyMap()
839 if (!ordering_.UseIsBeforeValueDefinition(*use, *b.value, dataflow_)) { in LiveRangeBefore()
940 const HloDataflowAnalysis& dataflow_; member in xla::__anon49efd3e50111::CopyRemover
Dbfloat16_propagation.h207 std::unique_ptr<HloDataflowAnalysis> dataflow_; variable