tensorflow_datasets

[cpu]
tensorflow

[gpu]
tensorflow-gpu
