Searched refs:RecordBufferEnqueue (Results 1 – 10 of 10) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/data/ |
D | parallel_batch_dataset_op.cc | 369 RecordBufferEnqueue(ctx.get(), result->output); in CallBatching() 458 RecordBufferEnqueue(ctx, result->output); in ReadBatchResult()
|
D | prefetch_dataset_op.cc | 296 RecordBufferEnqueue(ctx, buffer_element.value); in RestoreInternal() 502 RecordBufferEnqueue(ctx.get(), buffer_element.value); in PrefetchThread()
|
D | parallel_map_dataset_op.cc | 348 RecordBufferEnqueue(ctx, result.return_values); in RestoreInternal() 436 RecordBufferEnqueue(ctx.get(), result->return_values); in CallFunction()
|
D | window_dataset_op.cc | 177 RecordBufferEnqueue(ctx, element); in GetNextInternal()
|
D | cache_dataset_ops.cc | 791 RecordBufferEnqueue(ctx, *out_tensors); in GetNextInternal() 850 RecordBufferEnqueue(ctx, cache_->at(i)); in Initialize()
|
D | parallel_interleave_dataset_op.cc | 961 RecordBufferEnqueue(ctx_.get(), result->return_values); in ProcessElement() 1281 RecordBufferEnqueue(ctx, result->return_values); in ReadElement()
|
D | shuffle_dataset_op.cc | 212 this->RecordBufferEnqueue(ctx, input_element); in GetNextInternal()
|
/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
D | map_and_batch_dataset_op.cc | 520 RecordBufferEnqueue(ctx.get(), result->output); in EnsureOutputAllocated() 613 RecordBufferEnqueue(ctx, result->output); in ReadBatchResult()
|
D | parse_example_dataset_op.cc | 570 RecordBufferEnqueue(ctx.get(), result->return_values); in CallCompleted()
|
/external/tensorflow/tensorflow/core/framework/ |
D | dataset.h | 1047 void RecordBufferEnqueue(IteratorContext* ctx, in RecordBufferEnqueue() function
|