Searched refs:batch_prefix (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/data/ |
D | parallel_batch_dataset_op.cc | 443 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 …]
|
D | dataset_utils.cc | 712 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()
|
D | dataset_utils.h | 305 const string& batch_prefix, IteratorContext* ctx, 310 const string& iterator_prefix, const string& batch_prefix,
|
/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
D | map_and_batch_dataset_op.cc | 595 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 …]
|