Fix the broadcastable flags in case of broadcastable input.
Same fix for broadcastable to the GPU versions.