Add a use_partial_data to the data block api

[sgugger]

Thanks! This seems like a good and useful idea, will review later today.