Fix while_loop test.

- add output tensor to main block
- fix bug where tensor cannot serialize list of const data

Signed-off-by: Kevin Cheng <kevin.cheng@arm.com>
Change-Id: I40b96d634642e24e6fae6fa3ad6846f5911005f7