Commit 8133563
authored
[IO] Add
This patch introduces `hb.data.rebatch` for resizing batches from tabular data.
**Example**:
```python
import tensorflow as tf
import hybridbackend.tensorflow as hb
ds = hb.data.ParquetDataset(filename, micro_batch_size)
ds = ds.apply(hb.data.rebatch(batch_size))
```hb.data.rebatch for resizing batches from tabular data.1 parent d4ab633 commit 8133563
10 files changed
Lines changed: 1252 additions & 2 deletions
File tree
- docs/tutorials
- hybridbackend
- cpp/tensorflow/ops
- tensorflow/data
- tests/tensorflow/data
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
36 | 36 | | |
37 | 37 | | |
38 | 38 | | |
| 39 | + | |
39 | 40 | | |
40 | 41 | | |
41 | 42 | | |
| |||
89 | 90 | | |
90 | 91 | | |
91 | 92 | | |
| 93 | + | |
| 94 | + | |
| 95 | + | |
| 96 | + | |
| 97 | + | |
| 98 | + | |
| 99 | + | |
| 100 | + | |
| 101 | + | |
| 102 | + | |
| 103 | + | |
| 104 | + | |
| 105 | + | |
| 106 | + | |
| 107 | + | |
| 108 | + | |
| 109 | + | |
| 110 | + | |
| 111 | + | |
| 112 | + | |
92 | 113 | | |
93 | 114 | | |
94 | 115 | | |
| |||
0 commit comments