tf.lite.RepresentativeDataset