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