diff options
Diffstat (limited to 'tensorflow/core/kernels/data/parallel_map_iterator.h')
| -rw-r--r-- | tensorflow/core/kernels/data/parallel_map_iterator.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/core/kernels/data/parallel_map_iterator.h b/tensorflow/core/kernels/data/parallel_map_iterator.h index dc26c5cf25..813f13c9e4 100644 --- a/tensorflow/core/kernels/data/parallel_map_iterator.h +++ b/tensorflow/core/kernels/data/parallel_map_iterator.h @@ -30,7 +30,7 @@ namespace data { // 3. A `std::vector<Tensor>*` to which the function will write the result. // 4. A `StatusCallback` that should be invoked when the function is complete. using ParallelMapIteratorFunction = - std::function<void(IteratorContext*, std::vector<Tensor>, + std::function<void(IteratorContext*, const string&, std::vector<Tensor>, std::vector<Tensor>*, StatusCallback)>; // Returns a new iterator that applies `map_func` to the elements of |