Home
last modified time | relevance | path

Searched refs:aliased_buffers (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_alias_analysis.cc82 std::vector<BufferNumber> aliased_buffers = ComputeAliasedBuffers(*value); in MergeAliasedBuffers() local
85 if (aliased_buffers.empty()) { in MergeAliasedBuffers()
99 if (aliased_buffers.size() > 1) { in MergeAliasedBuffers()
100 for (int64 i = 1; i < aliased_buffers.size(); ++i) { in MergeAliasedBuffers()
101 MergeBuffers(/*from=*/aliased_buffers[i], in MergeAliasedBuffers()
102 /*to=*/aliased_buffers[0]); in MergeAliasedBuffers()
105 BufferNumber new_buffer = aliased_buffers[0]; in MergeAliasedBuffers()
176 const HloValue& value, std::vector<BufferNumber>* aliased_buffers) { in ComputeInputOutputAliasedBuffers() argument
197 aliased_buffers->push_back( in ComputeInputOutputAliasedBuffers()
206 aliased_buffers->push_back(GetBufferForValue(value)); in ComputeInputOutputAliasedBuffers()
[all …]
Dhlo_live_range.cc36 std::vector<const HloValue*> aliased_buffers; in NormalizeAliasedBuffers() local
39 aliased_buffers.push_back(hlo_value); in NormalizeAliasedBuffers()
43 aliased_buffers, [&](const HloValue* value1, const HloValue* value2) { in NormalizeAliasedBuffers()
51 for (int64 i = 0; i + 1 < aliased_buffers.size(); ++i) { in NormalizeAliasedBuffers()
52 const HloValue* value1 = aliased_buffers[i]; in NormalizeAliasedBuffers()
53 const HloValue* value2 = aliased_buffers[i + 1]; in NormalizeAliasedBuffers()