Searched refs:copy_element_fn (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | padded_batch_dataset_op.cc | 304 auto copy_element_fn = [component_index, &batch_elements, in GetNextInternal() local
325 [i, &status, &status_mu, &counter, ©_element_fn]() { in GetNextInternal()
326 Status s = copy_element_fn(i); in GetNextInternal()
334 status.Update(copy_element_fn(i)); in GetNextInternal()
|
| D | dataset_utils.cc | 872 auto copy_element_fn = [component_index, &batch_elements, in CopyBatch() local
898 [i, &status, &status_mu, &counter, ©_element_fn]() { in CopyBatch()
899 Status s = copy_element_fn(i); in CopyBatch()
907 status.Update(copy_element_fn(i)); in CopyBatch()
|