[neural network] Fix nn train without normalization & addresses memory leaks (#696)

* fixes order of warmUp() before getting inputs and output tensors

* fix memory leak - tensors only returned if returnTensors is true

Author: joeyklee