Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/data/
Ddataset_utils.h309 Status WriteBatch(int64 batch_size, int64 num_elements,
Dparallel_batch_dataset_op.cc479 TF_RETURN_IF_ERROR(WriteBatch(dataset()->batch_size_, in WriteBatchResult()
Ddataset_utils.cc745 Status WriteBatch(int64 batch_size, int64 num_elements, in WriteBatch() function
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dmap_and_batch_dataset_op.cc638 TF_RETURN_IF_ERROR(WriteBatch(dataset()->batch_size_, in WriteBatchResult()