Home
last modified time | relevance | path

Searched refs:choice_dataset (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/python/data/experimental/ops/
Dinterleave_ops.py235 def choose_from_datasets_v2(datasets, choice_dataset): argument
270 if not dataset_ops.get_structure(choice_dataset).is_compatible_with(
274 return _DirectedInterleaveDataset(choice_dataset, datasets)
278 def choose_from_datasets_v1(datasets, choice_dataset): argument
280 choose_from_datasets_v2(datasets, choice_dataset))
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Ddirected_interleave_dataset_test.py100 choice_dataset = dataset_ops.Dataset.from_tensor_slices(choice_array)
101 dataset = interleave_ops.choose_from_datasets(datasets, choice_dataset)
132 ], choice_dataset=dataset_ops.Dataset.from_tensors(1.0))
138 ], choice_dataset=dataset_ops.Dataset.from_tensors([1.0]))
/external/tensorflow/tensorflow/contrib/data/python/ops/
Dinterleave_ops.py165 def choose_from_datasets(datasets, choice_dataset): argument
200 return interleave_ops.choose_from_datasets(datasets, choice_dataset)
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.data.experimental.pbtxt101 argspec: "args=[\'datasets\', \'choice_dataset\'], varargs=None, keywords=None, defaults=None"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.data.experimental.pbtxt101 argspec: "args=[\'datasets\', \'choice_dataset\'], varargs=None, keywords=None, defaults=None"