Home
last modified time | relevance | path

Searched refs:RecordBufferEnqueue (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Dparallel_batch_dataset_op.cc369 RecordBufferEnqueue(ctx.get(), result->output); in CallBatching()
458 RecordBufferEnqueue(ctx, result->output); in ReadBatchResult()
Dprefetch_dataset_op.cc296 RecordBufferEnqueue(ctx, buffer_element.value); in RestoreInternal()
502 RecordBufferEnqueue(ctx.get(), buffer_element.value); in PrefetchThread()
Dparallel_map_dataset_op.cc348 RecordBufferEnqueue(ctx, result.return_values); in RestoreInternal()
436 RecordBufferEnqueue(ctx.get(), result->return_values); in CallFunction()
Dwindow_dataset_op.cc177 RecordBufferEnqueue(ctx, element); in GetNextInternal()
Dcache_dataset_ops.cc791 RecordBufferEnqueue(ctx, *out_tensors); in GetNextInternal()
850 RecordBufferEnqueue(ctx, cache_->at(i)); in Initialize()
Dparallel_interleave_dataset_op.cc961 RecordBufferEnqueue(ctx_.get(), result->return_values); in ProcessElement()
1281 RecordBufferEnqueue(ctx, result->return_values); in ReadElement()
Dshuffle_dataset_op.cc212 this->RecordBufferEnqueue(ctx, input_element); in GetNextInternal()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dmap_and_batch_dataset_op.cc520 RecordBufferEnqueue(ctx.get(), result->output); in EnsureOutputAllocated()
613 RecordBufferEnqueue(ctx, result->output); in ReadBatchResult()
Dparse_example_dataset_op.cc570 RecordBufferEnqueue(ctx.get(), result->return_values); in CallCompleted()
/external/tensorflow/tensorflow/core/framework/
Ddataset.h1047 void RecordBufferEnqueue(IteratorContext* ctx, in RecordBufferEnqueue() function