fix(datasets): Use put() and get() instead of copy in TensorFlowModelDataset's _save and _load methods.

fix(datasets): Use put() and get() instead of copy in TensorFlowModelDataset's _save and _load methods. #2335