Searched refs:dataflow_ (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_alias_analysis.cc | 64 : 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 …]
|
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() 616 dataflow_ in ResolveInconsistencyOfAliasingBuffersHelper() 642 dataflow_->GetValueSet(hlo, index).values()) { in ResolveInconsistencyOfAliasingBuffersHelper() 874 TF_ASSIGN_OR_RETURN(dataflow_, HloDataflowAnalysis::Run(*module)); in Run()
|
D | copy_insertion.cc | 489 : 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
|
D | bfloat16_propagation.h | 212 std::unique_ptr<HloDataflowAnalysis> dataflow_; variable
|