The results should be the same as test_small_model_tf.