Home
last modified time | relevance | path

Searched refs:allocation_callback (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/data/
Ddataset_utils.cc724 std::function<Status()> allocation_callback, in CopyBatch() argument
745 if (allocation_callback) { in CopyBatch()
746 TF_RETURN_IF_ERROR(allocation_callback()); in CopyBatch()
Ddataset_utils.h250 std::function<Status()> allocation_callback,
/external/tensorflow/tensorflow/core/kernels/data/
Dparallel_batch_dataset_op.cc400 auto allocation_callback = in CallBatching()
409 std::move(allocation_callback), &result->output); in CallBatching()