| 31 | } |
| 32 | |
| 33 | Stream Stream::batch( |
| 34 | int64_t batch_size, |
| 35 | const std::unordered_map<std::string, double>& pad_values, |
| 36 | const std::unordered_map<std::string, int>& batch_dims) const { |
| 37 | return Stream(std::make_shared<stream::Batch>( |
| 38 | self_, batch_size, pad_values, batch_dims)); |
| 39 | } |
| 40 | |
| 41 | Stream Stream::buffered( |
| 42 | int64_t buffer_size, |