Home
last modified time | relevance | path

Searched refs:BufferAlias (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dtuple_points_to_analysis.h158 class BufferAlias {
160 BufferAlias(HloInstruction* instruction, const ShapeIndex& index) in BufferAlias() function
167 bool operator==(const BufferAlias& other) const {
170 bool operator!=(const BufferAlias& other) const { return !(*this == other); }
179 std::ostream& operator<<(std::ostream& out, const BufferAlias& buffer_alias);
206 using BufferAliasVector = absl::InlinedVector<BufferAlias, 1>;
Dbuffer_liveness.cc102 for (const BufferAlias& alias : points_to_analysis_->GetBufferAliases(a)) { in live_range_strictly_before()
130 for (const BufferAlias& alias : points_to_analysis_->GetBufferAliases(a)) { in live_range_strictly_before()
Dtuple_points_to_analysis.cc38 string BufferAlias::ToString() const { in ToString()
43 std::ostream& operator<<(std::ostream& out, const BufferAlias& buffer_alias) { in operator <<()
567 for (const BufferAlias& alias : logical_buffer_aliases_.at(b->id())) { in ToString()
613 for (const BufferAlias& alias : GetBufferAliases(*buffer)) { in DoesNotUseOperandBuffer()
641 for (const BufferAlias& alias : GetBufferAliases(*buffer)) { in GetAllUsesOfInstructionAtIndex()
Dhlo.proto437 message BufferAlias { message
443 repeated BufferAlias buffer_aliases = 2;
Dtuple_points_to_analysis_test.cc111 std::vector<BufferAlias> expected_aliases; in ExpectHasBufferAliases()
113 expected_aliases.push_back(BufferAlias(pair.first, pair.second)); in ExpectHasBufferAliases()
Dbuffer_assignment.cc709 for (const BufferAlias& alias : in ToProto()
715 BufferAssignmentProto::BufferAlias* proto_alias = in ToProto()
Dhlo_rematerialization.cc297 for (const BufferAlias& buffer_alias : in GetUsers()