Lines Matching refs:dataset
42 """Test a dataset that represents a single tuple of tensors."""
45 iterator = (dataset_ops.Dataset.from_tensors(components)
67 """Test a dataset that represents a single tuple of tensors."""
78 dataset_ops.Dataset.from_tensors(components)
96 """Test an dataset that represents a single tuple of tensors."""
108 dataset_ops.Dataset.from_tensors(components)
130 """Test a dataset that represents the slices from a tuple of tensors."""
137 iterator = (dataset_ops.Dataset.from_tensor_slices(components)
155 """Test a dataset that represents the slices from a tuple of tensors."""
166 dataset_ops.Dataset.from_tensor_slices(components)
211 """Test a dataset that represents the slices from a tuple of tensors."""
225 dataset_ops.Dataset.from_tensor_slices(components)
276 iterator = (dataset_ops.Dataset.from_tensor_slices(components)
296 """Test a dataset based on slices of a `tf.SparseTensor`."""
298 iterator = (dataset_ops.Dataset.from_sparse_tensor_slices(st)
349 dataset = dataset_ops.Dataset.from_tensors(components)
351 dtypes.int64), dataset.output_types)
352 self.assertEquals(([3], ([2], [2]), [3]), dataset.output_shapes)
354 dataset = dataset.shuffle(10, 10)
356 dtypes.int64), dataset.output_types)
357 self.assertEquals(([3], ([2], [2]), [3]), dataset.output_shapes)
359 dataset = dataset.repeat(-1)
361 dtypes.int64), dataset.output_types)
362 self.assertEquals(([3], ([2], [2]), [3]), dataset.output_shapes)
364 dataset = dataset.filter(lambda x, y, z: True)
366 dtypes.int64), dataset.output_types)
367 self.assertEquals(([3], ([2], [2]), [3]), dataset.output_shapes)
369 dataset = dataset.take(5)
371 dtypes.int64), dataset.output_types)
372 self.assertEquals(([3], ([2], [2]), [3]), dataset.output_shapes)
374 dataset = dataset.map(lambda x, y, z: ((x, z), (y[0], y[1])))
376 (dtypes.float64, dtypes.float64)), dataset.output_types)
377 self.assertEquals((([3], [3]), ([2], [2])), dataset.output_shapes)
379 dataset = dataset.flat_map(
380 lambda x, y: dataset_ops.Dataset.from_tensors(((x[0], x[1]),
384 (dtypes.float64, dtypes.float64)), dataset.output_types)
385 self.assertEquals((([3], [3]), ([2], [2])), dataset.output_shapes)
387 dataset = dataset.batch(32)
389 (dtypes.float64, dtypes.float64)), dataset.output_types)
391 nest.pack_sequence_as(dataset.output_shapes, [
393 for s in nest.flatten(dataset.output_shapes)
396 iterator = dataset.make_one_shot_iterator()
407 iterator = dataset.make_initializable_iterator()
425 dataset = dataset_ops.Dataset.from_tensor_slices(components_for_slices)
427 dtypes.int64), dataset.output_types)
428 self.assertEquals(([], ([], []), []), dataset.output_shapes)
432 dataset = dataset_ops.Dataset.from_tensors(components)
433 self.assertEquals(dtypes.int32, dataset.output_types["a"]["aa"])
434 self.assertEquals(dtypes.float32, dataset.output_types["a"]["ab"])
435 self.assertEquals(dtypes.int32, dataset.output_types["b"])
436 self.assertEquals([], dataset.output_shapes["a"]["aa"])
437 self.assertEquals([2], dataset.output_shapes["a"]["ab"])
438 self.assertEquals([3], dataset.output_shapes["b"])
443 dataset = dataset_ops.Dataset.from_tensors(components)
444 self.assertEquals(dtypes.int64, dataset.output_types)
445 self.assertEquals([3], dataset.output_shapes)
447 dataset = dataset.filter(
449 self.assertEquals(dtypes.int64, dataset.output_types)
450 self.assertEquals([3], dataset.output_shapes)
452 dataset = dataset.map(lambda x: array_ops.stack([x, x]))
453 self.assertEquals(dtypes.int64, dataset.output_types)
454 self.assertEquals([2, 3], dataset.output_shapes)
456 dataset = dataset.flat_map(
457 lambda x: dataset_ops.Dataset.from_tensor_slices(x))
458 self.assertEquals(dtypes.int64, dataset.output_types)
459 self.assertEquals([3], dataset.output_shapes)
461 iterator = dataset.make_one_shot_iterator()
471 dataset = dataset_ops.Dataset.from_tensors(0)
480 dataset = dataset.map(lambda x: x + var_0.read_value())
485 dataset = dataset.map(lambda x: x + var_1.read_value())
488 iterator = dataset.make_initializable_iterator()
506 dataset = (
507 dataset_ops.Dataset.from_tensor_slices(input_data)
509 iterator = dataset.make_initializable_iterator()
543 dataset = (
544 dataset_ops.Dataset.from_tensor_slices(input_data)
546 iterator = dataset.make_initializable_iterator()
582 dataset = (
583 dataset_ops.Dataset.from_tensor_slices(input_data.reshape(100, 100))
585 iterator = dataset.make_initializable_iterator()
620 dataset = (
621 dataset_ops.Dataset.from_tensor_slices(input_data).batch(batch_size)
623 iterator = dataset.make_initializable_iterator()
