Searched refs:choice_dataset (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | interleave_ops.py | 243 def choose_from_datasets_v2(datasets, choice_dataset): argument
278 if not structure.are_compatible(choice_dataset.element_spec,
282 return _DirectedInterleaveDataset(choice_dataset, datasets)
286 def choose_from_datasets_v1(datasets, choice_dataset): argument
288 choose_from_datasets_v2(datasets, choice_dataset))
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | directed_interleave_dataset_test.py | 113 choice_dataset = dataset_ops.Dataset.from_tensor_slices(choice_array)
114 dataset = interleave_ops.choose_from_datasets(datasets, choice_dataset)
146 ], choice_dataset=dataset_ops.Dataset.from_tensors(1.0))
152 ], choice_dataset=dataset_ops.Dataset.from_tensors([1.0]))
157 choice_dataset=dataset_ops.Dataset.from_tensors(
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.data.experimental.pbtxt | 101 argspec: "args=[\'datasets\', \'choice_dataset\'], varargs=None, keywords=None, defaults=None"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.data.experimental.pbtxt | 129 argspec: "args=[\'datasets\', \'choice_dataset\'], varargs=None, keywords=None, defaults=None"
|