Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_rematerialization.cc106 bool IsSupportedIndirectUser(const HloInstruction* instruction) { in IsSupportedIndirectUser() function
465 !IsSupportedIndirectUser(buffer_alias.instruction())) { in GetUsers()
1115 if (!IsSupportedIndirectUser(user.user->instruction) || in AddRematerializedInstruction()
1159 if (IsSupportedIndirectUser(source_item->instruction)) { in AddRematerializedInstruction()
1647 if (!IsSupportedIndirectUser(u) || !uses_empty(u)) { in RematerializeInstructions()