Searched refs:GetUniqueBufferAt (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_alias_analysis_test.cc | 136 EXPECT_EQ(analysis.GetUniqueBufferAt(instruction).GetUniqueValue(), in TEST_F()
169 EXPECT_EQ(analysis.GetUniqueBufferAt(tuple, /*index=*/{}).GetUniqueValue(), in TEST_F()
171 EXPECT_EQ(analysis.GetUniqueBufferAt(tuple, /*index=*/{0}).GetUniqueValue(), in TEST_F()
173 EXPECT_EQ(analysis.GetUniqueBufferAt(tuple, /*index=*/{1}).GetUniqueValue(), in TEST_F()
178 EXPECT_EQ(analysis.GetUniqueBufferAt(param0), in TEST_F()
179 analysis.GetUniqueBufferAt(tuple, /*index=*/{0})); in TEST_F()
180 EXPECT_EQ(analysis.GetUniqueBufferAt(param0), in TEST_F()
181 analysis.GetUniqueBufferAt(gte0)); in TEST_F()
185 analysis.GetUniqueBufferAt(param0).ComputePositions(), in TEST_F()
211 analysis.GetUniqueBufferAt(param0).ComputePositions(), in TEST_F()
[all …]
|
| D | hlo_alias_analysis.h | 67 const HloBuffer& GetUniqueBufferAt(const HloInstruction* instruction,
69 HloBuffer& GetUniqueBufferAt(const HloInstruction* instruction,
|
| D | hlo_alias_analysis.cc | 381 const HloBuffer& HloAliasAnalysis::GetUniqueBufferAt( in GetUniqueBufferAt() function in xla::HloAliasAnalysis
388 HloBuffer& HloAliasAnalysis::GetUniqueBufferAt( in GetUniqueBufferAt() function in xla::HloAliasAnalysis
391 ->GetUniqueBufferAt(instruction, index) in GetUniqueBufferAt()
|
| D | memory_space_assignment.cc | 2943 const HloBuffer& buffer = alias_analysis->GetUniqueBufferAt( in ExportAndColorBuffers()
3323 alias_analysis->GetUniqueBufferAt(position.instruction, position.index); in VerifyAndExportHeapSimulatorTrace()
|
| D | buffer_assignment.cc | 1384 const HloBuffer& buffer = alias_analysis.GetUniqueBufferAt( in AssignPresetBuffers()
|
| D | memory_space_assignment_test.cc | 242 HloBuffer& buffer = alias_analysis->GetUniqueBufferAt(instruction, index);
|