Home
last modified time | relevance | path

Searched refs:DatasetV2 (Results 1 – 25 of 68) sorted by relevance

123

/external/tensorflow/tensorflow/python/autograph/operators/
Dpy_builtins_test.py59 dataset = dataset_ops.DatasetV2.from_tensor_slices([-1, 2, 3])
68 dataset_1 = dataset_ops.DatasetV2.from_tensor_slices([-1, 2, 3])
69 dataset_2 = dataset_ops.DatasetV2.from_tensor_slices([1, -2, 3])
70 dataset = dataset_ops.DatasetV2.zip((dataset_1, dataset_2))
79 dataset_1 = dataset_ops.DatasetV2.from_tensor_slices([-1, 2, 3])
80 dataset_2 = dataset_ops.DatasetV2.from_tensor_slices([1, -2, 3])
81 dataset_3 = dataset_ops.DatasetV2.from_tensor_slices([-1, -2, -3])
82 dataset_4 = dataset_ops.DatasetV2.zip((dataset_1, dataset_2))
83 dataset = dataset_ops.DatasetV2.zip((dataset_3, dataset_4))
128 dataset = dataset_ops.DatasetV2.from_tensor_slices([3, 2, 1])
[all …]
Dpy_builtins.py182 if isinstance(x, dataset_ops.DatasetV2):
250 if isinstance(s, dataset_ops.DatasetV2):
394 if isinstance(s, dataset_ops.DatasetV2):
412 if all(isinstance(x, dataset_ops.DatasetV2) for x in iterables):
418 return dataset_ops.DatasetV2.zip(iterables)
426 if all(isinstance(x, dataset_ops.DatasetV2) for x in iterables):
432 return dataset_ops.DatasetV2.zip(iterables).map(fn)
530 if isinstance(iterable, dataset_ops.DatasetV2):
544 if isinstance(iterable, dataset_ops.DatasetV2):
573 if isinstance(iterable, dataset_ops.DatasetV2):
/external/tensorflow/tensorflow/python/distribute/
Dinput_lib_type_spec_test.py65 dataset = dataset_ops.DatasetV2.range(10).batch(2)
98 dataset = dataset_ops.DatasetV2.range(10).batch(2)
142 dataset = dataset_ops.DatasetV2.range(10).batch(
183 ds = dataset_ops.DatasetV2.from_tensor_slices(
215 dataset = dataset_ops.DatasetV2.from_tensor_slices((a, b)).batch(2)
243 ds1 = dataset_ops.DatasetV2.range(10).batch(2).batch(5)
244 ds2 = dataset_ops.DatasetV2.from_tensors(
289 dataset = dataset_ops.DatasetV2.from_tensor_slices([fname1, fname2])
351 dataset = dataset_ops.DatasetV2.from_tensor_slices({
412 dataset = dataset_ops.DatasetV2.from_tensor_slices({
[all …]
Dstrategy_common_test.py203 d = dataset_ops.DatasetV2.range(100).repeat().batch(batch_size)
223 dataset = dataset_ops.DatasetV2.range(14).batch(
247 dataset = dataset_ops.DatasetV2.range(14).batch(
Ddistribute_lib_test.py58 return dataset_ops.DatasetV2.from_tensors(1.).repeat()
520 dataset_fn = lambda _: dataset_ops.DatasetV2.range(10).batch(2)
537 dataset_fn = lambda _: dataset_ops.DatasetV2.range(10).batch(2)
544 dataset_fn = lambda _: dataset_ops.DatasetV2.range(10).batch(2)
/external/tensorflow/tensorflow/python/keras/utils/
Ddataset_creator_test.py46 dataset_fn = lambda: dataset_ops.DatasetV2.from_tensor_slices([1, 1])
50 next(iter(dataset_ops.DatasetV2.from_tensor_slices([1, 1]))))
67 dataset = dataset_ops.DatasetV2.from_tensors(([1.], [1.])).repeat()
Ddataset_creator.py64 if not isinstance(dataset, dataset_ops.DatasetV2):
/external/tensorflow/tensorflow/python/data/ops/
Ddataset_ops.py111 class DatasetV2(collections_abc.Iterable, tracking_base.Trackable, class
851 generator_state = DatasetV2._GeneratorState(generator)
2020 if not isinstance(dataset, DatasetV2):
2302 class DatasetV1(DatasetV2):
2553 @functools.wraps(DatasetV2.from_tensors)
2555 return DatasetV1Adapter(DatasetV2.from_tensors(tensors))
2558 @functools.wraps(DatasetV2.from_tensor_slices)
2560 return DatasetV1Adapter(DatasetV2.from_tensor_slices(tensors))
2576 @functools.wraps(DatasetV2.from_generator)
2583 DatasetV2.from_generator(generator, output_types, output_shapes, args,
[all …]
Dreaders.py53 if isinstance(filenames, dataset_ops.DatasetV2):
70 filenames = dataset_ops.DatasetV2.from_tensor_slices(filenames)
303 class TFRecordDatasetV2(dataset_ops.DatasetV2):
/external/tensorflow/tensorflow/core/kernels/data/
Dshuffle_dataset_op.h52 class DatasetV2; variable
71 class DatasetV2; variable
Dshuffle_dataset_op.cc499 class ShuffleDatasetOp::DatasetV2 : public ShuffleDatasetBase { class in tensorflow::data::ShuffleDatasetOp
501 DatasetV2(OpKernelContext* ctx, const DatasetBase* input, int64 buffer_size, in DatasetV2() function in tensorflow::data::ShuffleDatasetOp::DatasetV2
510 ~DatasetV2() override { in ~DatasetV2()
709 new ShuffleDatasetOp::DatasetV2(ctx, input, buffer_size, count, manager, in MakeDataset()
802 class ShuffleAndRepeatDatasetOp::DatasetV2 : public ShuffleDatasetBase { class in tensorflow::data::ShuffleAndRepeatDatasetOp
804 DatasetV2(OpKernelContext* ctx, const DatasetBase* input, int64 buffer_size, in DatasetV2() function in tensorflow::data::ShuffleAndRepeatDatasetOp::DatasetV2
814 ~DatasetV2() override { in ~DatasetV2()
938 *output = new ShuffleAndRepeatDatasetOp::DatasetV2( in MakeDataset()
/external/tensorflow/tensorflow/python/keras/layers/preprocessing/
Dpreprocessing_stage.py57 (dataset_ops.DatasetV2, np.ndarray, ops.EagerTensor)):
62 if isinstance(data, dataset_ops.DatasetV2):
92 if isinstance(data, dataset_ops.DatasetV2):
Dcategory_encoding_distribution_test.py60 inp_dataset = dataset_ops.DatasetV2.from_tensor_slices(input_array)
Dcategory_crossing_distribution_test.py61 inp_dataset = dataset_ops.DatasetV2.from_tensor_slices(
/external/tensorflow/tensorflow/python/keras/distribute/
Dparameter_server_training_test.py285 return dataset_ops.DatasetV2.from_tensor_slices(
329 validation_data=dataset_ops.DatasetV2.from_tensor_slices([1, 1]))
336 self._model_fit(x=dataset_ops.DatasetV2.from_tensor_slices([1, 1]))
344 model.evaluate(x=dataset_ops.DatasetV2.from_tensor_slices([1, 1]))
352 model.predict(x=dataset_ops.DatasetV2.from_tensor_slices([1, 1]))
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_integration_test.py159 dataset = dataset_ops.DatasetV2.range(4).map(map_fn).batch(batch_size)
177 eval_dataset = dataset_ops.DatasetV2.range(4).map(map_fn).batch(batch_size)
183 pred_dataset = dataset_ops.DatasetV2.range(4)
Dbase_preprocessing_layer_v1.py105 if not isinstance(data, (dataset_ops.DatasetV2, np.ndarray, ops.Tensor,
111 if isinstance(data, dataset_ops.DatasetV2):
Ddata_adapter.py303 indices_dataset = dataset_ops.DatasetV2.range(1)
343 flat_dataset = dataset_ops.DatasetV2.from_tensor_slices(first_k_indices)
345 index_remainder = dataset_ops.DatasetV2.from_tensors(array_ops.slice(
380 dataset = dataset_ops.DatasetV2.zip((
382 dataset_ops.DatasetV2.from_tensors(inputs).repeat()
565 not isinstance(v, (dataset_ops.DatasetV2,
601 dataset = dataset_ops.DatasetV2.from_tensor_slices(inputs)
711 return (isinstance(x, (dataset_ops.DatasetV1, dataset_ops.DatasetV2)) or
845 dataset = dataset_ops.DatasetV2.from_generator(
1630 dataset = dataset_ops.DatasetV2.from_tensors(data)
Dtraining_arrays_v1.py135 (dataset_ops.DatasetV1, dataset_ops.DatasetV2))
190 if isinstance(val_inputs, (dataset_ops.DatasetV1, dataset_ops.DatasetV2)):
500 if isinstance(inputs, (dataset_ops.DatasetV1, dataset_ops.DatasetV2)):
520 if isinstance(inputs, (dataset_ops.DatasetV1, dataset_ops.DatasetV2,
Dtraining_v1.py1709 dataset_ops.DatasetV2,
1751 if isinstance(x, (dataset_ops.DatasetV2, iterator_ops.Iterator,
2148 if isinstance(x, dataset_ops.DatasetV2):
2200 assert isinstance(x, dataset_ops.DatasetV2)
2275 if isinstance(x, (dataset_ops.DatasetV1, dataset_ops.DatasetV2)):
2361 if not isinstance(x, (dataset_ops.DatasetV1, dataset_ops.DatasetV2)):
2374 if isinstance(x, dataset_ops.DatasetV2):
2467 dataset_ops.DatasetV2)):
2480 if isinstance(inputs, (dataset_ops.DatasetV1, dataset_ops.DatasetV2)):
2516 if isinstance(orig_inputs, (dataset_ops.DatasetV1, dataset_ops.DatasetV2,
[all …]
/external/tensorflow/tensorflow/python/distribute/coordinator/
Dcluster_coordinator_test.py480 return dataset_ops.DatasetV2.range(1, 2)
506 return dataset_ops.DatasetV2.from_tensor_slices([2] * 10)
555 return dataset_ops.DatasetV2.range(0, 10).map(map_fn)
583 return dataset_ops.DatasetV2.range(0, 100).shuffle(100)
662 dataset = dataset_ops.DatasetV2.range(1, 10)
685 return dataset_ops.DatasetV2.range(1, 10)
959 dataset = dataset_ops.DatasetV2.range(1, 2)
977 lambda _: dataset_ops.DatasetV2.range(1, 2))
992 dataset = dataset_ops.DatasetV2.range(1, 2)
997 lambda _: dataset_ops.DatasetV2.range(1, 2))
/external/tensorflow/tensorflow/python/data/experimental/ops/
Dget_single_element.py62 if not isinstance(dataset, dataset_ops.DatasetV2):
Dinterleave_ops.py104 class _DirectedInterleaveDataset(dataset_ops.DatasetV2):
176 if not isinstance(weights, dataset_ops.DatasetV2):
Dwriters.py105 if not isinstance(dataset, dataset_ops.DatasetV2):
Doptimization.py69 class _ChooseFastestDataset(dataset_ops.DatasetV2):

123