[XLA] Restrict kMap inlining to elementwise non-fusion instructions

For instructions that cannot take arbitrary shapes changing the operand to a
different shape will create invalid HLO.

This is a bit awkward because XLA doesn't have a clear concept of "instruction
that can process any shape", but I think kLoop fusion is the only elementwise
op where this isn't true.

PiperOrigin-RevId: 236279529