Searched defs:batch_prefix (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/ |
| H A D | parallel_batch_dataset_op.cc | 529 string batch_prefix = strings::StrCat(kBatchResults, "_", index); in ReadBatchResult() local
555 string batch_prefix = strings::StrCat(kBatchResults, "_", index); in WriteBatchResult() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| H A D | map_and_batch_dataset_op.cc | 606 string batch_prefix = strings::StrCat(kBatchResults, "_", index); in ReadBatchResult() local
633 string batch_prefix = strings::StrCat(kBatchResults, "_", index); in WriteBatchResult() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/data/ |
| H A D | dataset_utils.cc | 607 const string& batch_prefix, std::vector<Tensor>* batch) { in ReadBatch()
639 const string& iterator_prefix, const string& batch_prefix, in WriteBatch()
|