Searched defs:TakeDataset (Results 1 – 3 of 3) sorted by relevance
27 TakeDataset(OpKernelContext* ctx, int64 count, const DatasetBase* input) in TakeDataset() function32 TakeDataset(DatasetContext::Params params, int64 count, in TakeDataset() function
2713 class TakeDataset(UnaryUnchangedStructureDataset): class
20600 func TakeDataset(scope *Scope, input_dataset tf.Output, count tf.Output, output_types []tf.DataType… func