Home
last modified time | relevance | path

Searched refs:batch_prefix (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Dparallel_batch_dataset_op.cc443 string batch_prefix = strings::StrCat(kBatchResults, "_", index); in ReadBatchResult() local
446 full_name(strings::StrCat(batch_prefix, "_", kEndOfInput))); in ReadBatchResult()
448 full_name(strings::StrCat(batch_prefix, "_", kNumElements)), in ReadBatchResult()
451 full_name(strings::StrCat(batch_prefix, "_", kCallFinished))); in ReadBatchResult()
454 batch_prefix, ctx, reader, &result->output)); in ReadBatchResult()
456 strings::StrCat(batch_prefix, "_", kStatus), in ReadBatchResult()
465 string batch_prefix = strings::StrCat(kBatchResults, "_", index); in WriteBatchResult() local
469 full_name(strings::StrCat(batch_prefix, "_", kEndOfInput)), "")); in WriteBatchResult()
472 full_name(strings::StrCat(batch_prefix, "_", kNumElements)), in WriteBatchResult()
476 full_name(strings::StrCat(batch_prefix, "_", kCallFinished)), "")); in WriteBatchResult()
[all …]
Ddataset_utils.cc712 const string& batch_prefix, IteratorContext* ctx, in ReadBatch() argument
717 strings::StrCat(batch_prefix, "_", kOutputSize)), in ReadBatch()
724 strings::StrCat(batch_prefix, "_", kOutput, "_", i)), in ReadBatch()
746 const string& iterator_prefix, const string& batch_prefix, in WriteBatch() argument
750 strings::StrCat(batch_prefix, "_", kOutputSize)), in WriteBatch()
759 strings::StrCat(batch_prefix, "_", kOutput, "_", i)), in WriteBatch()
764 strings::StrCat(batch_prefix, "_", kOutput, "_", i)), in WriteBatch()
Ddataset_utils.h305 const string& batch_prefix, IteratorContext* ctx,
310 const string& iterator_prefix, const string& batch_prefix,
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dmap_and_batch_dataset_op.cc595 string batch_prefix = strings::StrCat(kBatchResults, "_", index); in ReadBatchResult() local
598 full_name(strings::StrCat(batch_prefix, "_", kEndOfInput))); in ReadBatchResult()
600 full_name(strings::StrCat(batch_prefix, "_", kNumCalls)), in ReadBatchResult()
603 full_name(strings::StrCat(batch_prefix, "_", kNumElements)), in ReadBatchResult()
606 full_name(strings::StrCat(batch_prefix, "_", kOutputAllocated))); in ReadBatchResult()
609 batch_prefix, ctx, reader, &result->output)); in ReadBatchResult()
611 strings::StrCat(batch_prefix, "_", kStatus), in ReadBatchResult()
620 string batch_prefix = strings::StrCat(kBatchResults, "_", index); in WriteBatchResult() local
624 full_name(strings::StrCat(batch_prefix, "_", kEndOfInput)), "")); in WriteBatchResult()
627 full_name(strings::StrCat(batch_prefix, "_", kNumCalls)), in WriteBatchResult()
[all …]