Home
last modified time | relevance | path

Searched refs:dataflow_analysis_ (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_dataflow_analysis_test.cc2074 dataflow_analysis_ = HloDataflowAnalysis::Run( in RunAnalysis()
2087 std::unique_ptr<HloDataflowAnalysis> dataflow_analysis_; member in xla::__anon587423c40111::HloDataflowAnalysisTestBase
2109 EXPECT_TRUE(dataflow_analysis_->DoesNotUseOperandBuffer(tuple, {0}, gte0)); in TEST_F()
2110 EXPECT_TRUE(dataflow_analysis_->DoesNotUseOperandBuffer(tuple, {1}, gte1)); in TEST_F()
2111 EXPECT_FALSE(dataflow_analysis_->DoesNotUseOperandBuffer(tuple, {}, gte0)); in TEST_F()
2112 EXPECT_FALSE(dataflow_analysis_->DoesNotUseOperandBuffer(tuple, {}, gte1)); in TEST_F()
2145 EXPECT_TRUE(dataflow_analysis_->DoesNotUseOperandBuffer(tuple, {0}, fusion)); in TEST_F()
2146 EXPECT_FALSE(dataflow_analysis_->DoesNotUseOperandBuffer(tuple, {1}, fusion)); in TEST_F()
2188 EXPECT_TRUE(dataflow_analysis_->DoesNotUseOperandBuffer(tuple, {0}, fusion)); in TEST_F()
2189 EXPECT_FALSE(dataflow_analysis_->DoesNotUseOperandBuffer(tuple, {1}, fusion)); in TEST_F()
[all …]
Dhlo_alias_analysis.h84 HloDataflowAnalysis& dataflow_analysis() const { return *dataflow_analysis_; } in dataflow_analysis()
135 std::unique_ptr<HloDataflowAnalysis> dataflow_analysis_; variable
Dmemory_space_propagation.cc24 dataflow_analysis_ = std::move(dataflow_analysis); in Run()
60 const HloValue& value = dataflow_analysis_->GetUniqueValueAt( in Propagate()
Dmemory_space_propagation.h40 std::unique_ptr<HloDataflowAnalysis> dataflow_analysis_; variable
Dhlo_alias_analysis.cc399 dataflow_analysis_->GetValueSet(instruction, index).values()) { in ComputeBuffersAt()
413 dataflow_analysis_->GetInstructionValueSet(instruction)) { in InstructionBuffersAreAmbiguous()
431 dataflow_analysis_->GetInstructionValueSet(instruction)) { in InstructionBuffersAreDistinct()
532 TF_ASSIGN_OR_RETURN(alias_analysis->dataflow_analysis_, in Run()