Clean input argument rank limitations

Comparison ops, similar to the other elementwise ops, can operate
on tensors of any rank.

For those operators with rank limitations, make them consistent and
remove "dims".

Signed-off-by: Eric Kunze <eric.kunze@arm.com>
Change-Id: I33eb00122cec4c6625f080f9ed640f1d874fae87