Home
last modified time | relevance | path

Searched refs:GetValueDefinedAt (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_ordering_test.cc190 EXPECT_TRUE(ordering.IsDefinedBefore(dataflow->GetValueDefinedAt(param), in TEST_F()
191 dataflow->GetValueDefinedAt(constant))); in TEST_F()
192 EXPECT_TRUE(!ordering.IsDefinedBefore(dataflow->GetValueDefinedAt(constant), in TEST_F()
193 dataflow->GetValueDefinedAt(param))); in TEST_F()
244 EXPECT_TRUE(ordering.IsDefinedBefore(dataflow->GetValueDefinedAt(constant), in TEST_F()
245 dataflow->GetValueDefinedAt(xla_while))); in TEST_F()
247 dataflow->GetValueDefinedAt(constant), in TEST_F()
248 dataflow->GetValueDefinedAt(xla_while), *dataflow)); in TEST_F()
252 dataflow->GetValueDefinedAt(constant), in TEST_F()
253 dataflow->GetValueDefinedAt(convert), *dataflow)); in TEST_F()
[all …]
Dhlo_dataflow_analysis_test.cc81 return ordering.MayInterfere(analysis_->GetValueDefinedAt(a), in InstructionsMayInterfere()
82 analysis_->GetValueDefinedAt(b), *analysis_); in InstructionsMayInterfere()
127 EXPECT_THAT(analysis.GetValueDefinedAt(constant1).positions(), in TEST_P()
129 EXPECT_THAT(analysis.GetValueDefinedAt(constant2).positions(), in TEST_P()
131 EXPECT_THAT(analysis.GetValueDefinedAt(add).positions(), in TEST_P()
135 EXPECT_THAT(analysis.GetValueDefinedAt(constant1).uses(), in TEST_P()
137 EXPECT_THAT(analysis.GetValueDefinedAt(constant2).uses(), in TEST_P()
139 EXPECT_TRUE(analysis.GetValueDefinedAt(add).uses().empty()); in TEST_P()
142 EXPECT_FALSE(analysis.GetValueDefinedAt(constant1).live_out_of_module()); in TEST_P()
143 EXPECT_FALSE(analysis.GetValueDefinedAt(constant2).live_out_of_module()); in TEST_P()
[all …]
Dhlo_alias_analysis_test.cc84 const HloValue& GetValueDefinedAt(const HloInstruction* instruction, in GetValueDefinedAt() function in xla::__anona4abf5fc0111::HloAliasAnalysisTest
86 return analysis_->dataflow_analysis().GetValueDefinedAt(instruction, index);
137 GetValueDefinedAt(instruction)); in TEST_F()
170 GetValueDefinedAt(tuple, /*index=*/{})); in TEST_F()
172 GetValueDefinedAt(param0)); in TEST_F()
174 GetValueDefinedAt(param1)); in TEST_F()
383 UnorderedElementsAre(GetValueDefinedAt(param, {1}), in TEST_F()
384 GetValueDefinedAt(xla_while, /*index=*/{1}), in TEST_F()
385 GetValueDefinedAt(body_param, {1}), in TEST_F()
386 GetValueDefinedAt(cond_param, {1}), in TEST_F()
[all …]
Dhlo_dataflow_analysis.h96 const HloValue& GetValueDefinedAt(const HloInstruction* instruction,
98 HloValue& GetValueDefinedAt(const HloInstruction* instruction,
Dhlo_dataflow_analysis.cc132 const HloValue& HloDataflowAnalysis::GetValueDefinedAt( in GetValueDefinedAt() function in xla::HloDataflowAnalysis
138 HloValue& HloDataflowAnalysis::GetValueDefinedAt( in GetValueDefinedAt() function in xla::HloDataflowAnalysis
1206 GetValueDefinedAt(fusion_param, use.operand_index); in DoesNotUseOperandBuffer()
1304 GetValueDefinedAt(fusion_param, operand_index); in CanShareOperandBufferWithUser()
1367 const auto& uses = GetValueDefinedAt(operand, operand_index).uses(); in CanShareOperandBufferWithUser()
Dmemory_space_assignment_test.cc5143 dataflow_analysis->GetValueDefinedAt( in TEST_P()
5203 dataflow_analysis->GetValueDefinedAt( in TEST_P()