Searched refs:buffer_alias (Results 1 – 4 of 4) sorted by relevance
457 for (const BufferAlias& buffer_alias : in GetUsers() local459 for (const HloInstruction* user : buffer_alias.instruction()->users()) { in GetUsers()461 buffer_alias.instruction(), buffer_alias.index(), user)) { in GetUsers()468 if (buffer_alias.instruction() != logical_buffer->instruction() && in GetUsers()469 !IsSupportedIndirectUser(buffer_alias.instruction())) { in GetUsers()479 for (int64_t op_idx : user->OperandIndices(buffer_alias.instruction())) { in GetUsers()
179 std::ostream& operator<<(std::ostream& out, const BufferAlias& buffer_alias);
46 std::ostream& operator<<(std::ostream& out, const BufferAlias& buffer_alias) { in operator <<() argument47 out << buffer_alias.ToString(); in operator <<()
1460 for (const auto& buffer_alias : aliases) { in GetArrayUsesOfBuffer() local1461 if (!buffer_alias.instruction()->shape().IsArray()) { in GetArrayUsesOfBuffer()1466 CHECK(buffer_alias.index().empty()); in GetArrayUsesOfBuffer()1469 for (const HloInstruction* user : buffer_alias.instruction()->users()) { in GetArrayUsesOfBuffer()1471 user->OperandIndices(buffer_alias.instruction())) { in GetArrayUsesOfBuffer()