Update tests to use tf.data.generator (#168)
Update tests to use tf.data.generator (#168)

INTERNAL
* save

* save

tensorflow/tfjs,tensorflow/tfjs,tensorflow/tfjs-data,tensorflow/tfjs-data,tensorflow/tfjs,tensorflow/tfjs,tensorflow/tfjs,tensorflow/tfjs-data,tensorflow/tfjs,tensorflow/tfjs,tensorflow/tfjs,tensorflow/tfjs,tensorflow/tfjs-data