Searched refs:rotate_with_padding (Results 1 – 1 of 1) sorted by relevance
274 auto rotate_with_padding = [&](int64_t rotate_amount) { in HandleCustomCallSPMDInternal_RotateRight() local327 HloInstruction* rotated0 = rotate_with_padding(amount); in HandleCustomCallSPMDInternal_RotateRight()343 HloInstruction* rotated1 = rotate_with_padding( in HandleCustomCallSPMDInternal_RotateRight()