Convert tests to use tf instead of caffe

[ibreschi]

We should remove the Caffe dependencies from the tests.
Since most of our networks use TensorFlow, it's better to use it instead of Caffe in the tests.