Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_ordering_test.cc191 EXPECT_TRUE(ordering.IsDefinedBefore(dataflow->GetValueDefinedAt(param), in TEST_F()
192 dataflow->GetValueDefinedAt(constant))); in TEST_F()
193 EXPECT_TRUE(!ordering.IsDefinedBefore(dataflow->GetValueDefinedAt(constant), in TEST_F()
194 dataflow->GetValueDefinedAt(param))); in TEST_F()
245 EXPECT_TRUE(ordering.IsDefinedBefore(dataflow->GetValueDefinedAt(constant), in TEST_F()
246 dataflow->GetValueDefinedAt(xla_while))); in TEST_F()
248 dataflow->GetValueDefinedAt(constant), in TEST_F()
249 dataflow->GetValueDefinedAt(xla_while), *dataflow)); in TEST_F()
250 EXPECT_TRUE(ordering.MayInterfere(dataflow->GetValueDefinedAt(constant), in TEST_F()
251 dataflow->GetValueDefinedAt(xla_while), in TEST_F()
[all …]
Dhlo_dataflow_analysis_test.cc78 return ordering.MayInterfere(analysis_->GetValueDefinedAt(a), in InstructionsMayInterfere()
79 analysis_->GetValueDefinedAt(b), *analysis_); in InstructionsMayInterfere()
124 EXPECT_THAT(analysis.GetValueDefinedAt(constant1).positions(), in TEST_P()
126 EXPECT_THAT(analysis.GetValueDefinedAt(constant2).positions(), in TEST_P()
128 EXPECT_THAT(analysis.GetValueDefinedAt(add).positions(), in TEST_P()
132 EXPECT_THAT(analysis.GetValueDefinedAt(constant1).uses(), in TEST_P()
134 EXPECT_THAT(analysis.GetValueDefinedAt(constant2).uses(), in TEST_P()
136 EXPECT_TRUE(analysis.GetValueDefinedAt(add).uses().empty()); in TEST_P()
139 EXPECT_FALSE(analysis.GetValueDefinedAt(constant1).live_out_of_module()); in TEST_P()
140 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::__anon8f906bde0111::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()
391 UnorderedElementsAre(GetValueDefinedAt(param, {1}), in TEST_F()
392 GetValueDefinedAt(xla_while, /*index=*/{1}), in TEST_F()
393 GetValueDefinedAt(body_param, {1}), in TEST_F()
394 GetValueDefinedAt(cond_param, {1}), in TEST_F()
[all …]
Dhlo_dataflow_analysis.h92 const HloValue& GetValueDefinedAt(const HloInstruction* instruction,
94 HloValue& GetValueDefinedAt(const HloInstruction* instruction,
Dhlo_dataflow_analysis.cc87 const HloValue& HloDataflowAnalysis::GetValueDefinedAt( in GetValueDefinedAt() function in xla::HloDataflowAnalysis
93 HloValue& HloDataflowAnalysis::GetValueDefinedAt( in GetValueDefinedAt() function in xla::HloDataflowAnalysis
937 GetValueDefinedAt(fusion_param, use.operand_index); in DoesNotUseOperandBuffer()
1041 GetValueDefinedAt(fusion_param, operand_index); in CanShareOperandBufferWithUser()
1119 const auto& uses = GetValueDefinedAt(operand, operand_index).uses(); in CanShareOperandBufferWithUser()