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