prepare\_dataset\_shard ======================= .. currentmodule:: ray.train.tensorflow .. autofunction:: prepare_dataset_shard