| 9f4118d00fa9eb85f81a4eb3f96a5583ae5afcdc |
|
18-Jan-2018 |
Rohan Jain <rohanj@google.com> |
Changes the CapturedFunction implementation to not create a new FunctionLibraryRuntime etc. and just re-use the existing one provided. This allows for Dataset ops to run cross device and cross process.
PiperOrigin-RevId: 182292325
/external/tensorflow/tensorflow/core/kernels/data/parallel_interleave_dataset_op.cc
|
| 76db97fe3961651617371902a1a623df61f9ed81 |
|
20-Dec-2017 |
Brennan Saeta <saeta@google.com> |
Add prefetching into parallel_interleave
This change adds 2 parameters to parallel_interleave:
- prefetch_input_elements: determines the number of iterators to prefetch
allowing buffers to warm up and data to be pre-fetched without blocking
the main thread (i.e. the GetNext() call).
- buffer_output_elements: in order to avoid creating thousands of threads, we
fuse in the .prefetch() operator as an additional parameter. The value of
this parameter is identical to the value passed to `.prefetch()`
PiperOrigin-RevId: 179726088
/external/tensorflow/tensorflow/core/kernels/data/parallel_interleave_dataset_op.cc
|
| a5b2a0c9a3335d10c4dd3dfdff96149f74a4d120 |
|
15-Dec-2017 |
Jiri Simsa <jsimsa@google.com> |
Moving tf.data kernels to their own package.
PiperOrigin-RevId: 179112798
/external/tensorflow/tensorflow/core/kernels/data/parallel_interleave_dataset_op.cc
|