Change from assignment to a check for output dims for tensor ops

Output dimensions should have been worked out correctly by the end
of the compilation to TOSA

Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: I0193958d7d51bf1c6915a0953daa514f86516710