Searched refs:IsInstructionSafeForShmemTranspose (Results 1 – 1 of 1) sorted by relevance
3308 bool IsInstructionSafeForShmemTranspose(const HloInstruction* hlo) { in IsInstructionSafeForShmemTranspose() function3311 return IsInstructionSafeForShmemTranspose(user); in IsInstructionSafeForShmemTranspose()3328 return IsInstructionSafeForShmemTranspose(user); in IsInstructionSafeForShmemTranspose()3353 if (IsInstructionSafeForShmemTranspose(input)) { in FilterInputsForShmemTranspose()