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.cc93 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()
616 dataflow_ in ResolveInconsistencyOfAliasingBuffersHelper()
642 dataflow_->GetValueSet(hlo, index).values()) { in ResolveInconsistencyOfAliasingBuffersHelper()
874 TF_ASSIGN_OR_RETURN(dataflow_, HloDataflowAnalysis::Run(*module)); in Run()
Dcopy_insertion.cc489 : dataflow_(alias_analysis.dataflow_analysis()), ordering_(ordering) { in CopyRemover()
514 dataflow_) || in CopyRemover()
516 dataflow_)) in CopyRemover()
589 dataflow_.GetValueSet(instruction->operand(0)); in CreateCopyMap()
593 value_to_node.at(&dataflow_.GetUniqueValueAt(instruction)); in CreateCopyMap()
863 return ordering_.UsesBeforeValueDefinition(a.uses, *b.value, dataflow_); in LiveRangeBefore()
958 const HloDataflowAnalysis& dataflow_; member in xla::__anond6a214430111::CopyRemover
Dbfloat16_propagation.h212 std::unique_ptr<HloDataflowAnalysis> dataflow_; variable