Home
last modified time | relevance | path

Searched refs:buffer_alias (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_rematerialization.cc297 for (const BufferAlias& buffer_alias : in GetUsers() local
299 for (const HloInstruction* user : buffer_alias.instruction()->users()) { in GetUsers()
301 buffer_alias.instruction(), buffer_alias.index(), user)) { in GetUsers()
308 if (buffer_alias.instruction() != logical_buffer->instruction()) { in GetUsers()
Dtuple_points_to_analysis.h179 std::ostream& operator<<(std::ostream& out, const BufferAlias& buffer_alias);
Dtuple_points_to_analysis.cc43 std::ostream& operator<<(std::ostream& out, const BufferAlias& buffer_alias) { in operator <<() argument
44 out << buffer_alias.ToString(); in operator <<()
Dlayout_assignment.cc1173 for (const auto& buffer_alias : points_to_analysis.GetBufferAliases(buffer)) { in GetArrayUsesOfBuffer() local
1174 if (!buffer_alias.instruction()->shape().IsArray()) { in GetArrayUsesOfBuffer()
1179 CHECK(buffer_alias.index().empty()); in GetArrayUsesOfBuffer()
1182 for (const HloInstruction* user : buffer_alias.instruction()->users()) { in GetArrayUsesOfBuffer()
1184 user->OperandIndices(buffer_alias.instruction())) { in GetArrayUsesOfBuffer()