Searched defs:TakeDataset (Results 1 – 3 of 3) sorted by relevance
35 TakeDataset::TakeDataset(OpKernelContext* ctx, int64 count, in TakeDataset() function in tensorflow::data::TakeDataset41 TakeDataset::TakeDataset(DatasetContext::Params params, int64 count, in TakeDataset() function in tensorflow::data::TakeDataset
3908 class TakeDataset(UnaryUnchangedStructureDataset): class
21091 func TakeDataset(scope *Scope, input_dataset tf.Output, count tf.Output, output_types []tf.DataType… func