Home
last modified time | relevance | path

Searched refs:reserve_size_ (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Dbatch_dataset_op.cc58 reserve_size_(drop_remainder ? batch_size in Dataset()
169 batch_elements.reserve(dataset()->reserve_size_); in GetNextInternal()
247 const int64 reserve_size_; member in tensorflow::data::BatchDatasetOp::Dataset
Dparallel_batch_dataset_op.cc74 reserve_size_(drop_remainder ? batch_size in Dataset()
334 batch_elements->reserve(dataset()->reserve_size_); in CallBatching()
516 const int64 reserve_size_; member in tensorflow::data::ParallelBatchDatasetOp::Dataset