Prevent collapsing batch dims in dot ops with constants #19540

Job	Run time
pyflakes	3m 49s
Check if image exists in registry	4s
licensing	3m 18s
cancel	4s
linux (ubuntu-20.04, debug)	25m 36s
linux-fpga (ubuntu-20.04, debug)	27m 31s
linux (ubuntu-20.04, release)	3m 49s
linux (ubuntu-20.04, codecov)	5m 15s
Build SLES image	0s
Build image	0s
sles	2m 29s
tidy	4m 17s
cppcheck	43m 53s
format	4m 17s
	2h 4m 22s

Provide feedback