Searched refs:dataset_split (Results 1 – 2 of 2) sorted by relevance
47 def get_input_fn(dataset_split, argument56 tensors=[dataset_split.images,57 dataset_split.labels.astype(np.int32)],
55 def get_input_fn(dataset_split, batch_size, capacity=10000, min_after_dequeue=3000):59 tensors=[dataset_split.images, dataset_split.labels.astype(np.int32)],