Searched defs:BatchDataset (Results 1 – 3 of 3) sorted by relevance
31 TEST(ParallelBatch, BatchDataset) { in TEST() argument
5091 class BatchDataset(UnaryDataset): class
2430 func BatchDataset(scope *Scope, input_dataset tf.Output, batch_size tf.Output, output_types []tf.Da… func