| /aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| H A D | directed_interleave_dataset_test.py | 211 choice_dataset = dataset_ops.Dataset.from_tensor_slices(choice_array)
212 dataset = dataset_ops.Dataset.choose_from_datasets(datasets, choice_dataset)
227 choice_dataset = dataset_ops.Dataset.from_tensor_slices(choice_array)
229 datasets, choice_dataset, stop_on_empty_dataset=True)
240 choice_dataset = dataset_ops.Dataset.from_tensor_slices(choice_array)
242 datasets, choice_dataset, stop_on_empty_dataset=False)
257 choice_dataset=dataset_ops.Dataset.range(0),
265 choice_dataset = dataset_ops.Dataset.range(2).repeat(5)
266 ds = dataset_ops.Dataset.choose_from_datasets([ds1, ds2], choice_dataset)
304 choice_dataset=dataset_ops.Dataset.from_tensors(1.0))
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/ops/ |
| H A D | interleave_ops.py | 177 choice_dataset, argument
222 choice_dataset=choice_dataset,
234 choice_dataset, argument
237 choose_from_datasets_v2(datasets, choice_dataset, stop_on_empty_dataset))
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/service/ |
| H A D | dynamic_sharding_test.py | 282 choice_dataset = dataset_ops.Dataset.from_tensor_slices(choice_array)
283 ds = dataset_ops.Dataset.choose_from_datasets(datasets, choice_dataset)
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/ops/ |
| H A D | dataset_ops.py | 3560 choice_dataset, argument
3605 if not isinstance(choice_dataset, DatasetV2):
3608 if not structure.are_compatible(choice_dataset.element_spec,
3616 choice_dataset = _apply_rewrite(choice_dataset, "replicate_on_split")
3618 return _DirectedInterleaveDataset(choice_dataset, datasets,
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/ |
| H A D | tensorflow.data.experimental.-csv-dataset.pbtxt | 41 …argspec: "args=[\'datasets\', \'choice_dataset\', \'stop_on_empty_dataset\'], varargs=None, keywor…
|
| H A D | tensorflow.data.-fixed-length-record-dataset.pbtxt | 41 …argspec: "args=[\'datasets\', \'choice_dataset\', \'stop_on_empty_dataset\'], varargs=None, keywor…
|
| H A D | tensorflow.data.-dataset.pbtxt | 39 …argspec: "args=[\'datasets\', \'choice_dataset\', \'stop_on_empty_dataset\'], varargs=None, keywor…
|
| H A D | tensorflow.data.experimental.-sql-dataset.pbtxt | 41 …argspec: "args=[\'datasets\', \'choice_dataset\', \'stop_on_empty_dataset\'], varargs=None, keywor…
|
| H A D | tensorflow.data.experimental.-random-dataset.pbtxt | 42 …argspec: "args=[\'datasets\', \'choice_dataset\', \'stop_on_empty_dataset\'], varargs=None, keywor…
|
| H A D | tensorflow.experimental.dtensor.-d-tensor-dataset.pbtxt | 42 …argspec: "args=[\'datasets\', \'choice_dataset\', \'stop_on_empty_dataset\'], varargs=None, keywor…
|
| H A D | tensorflow.data.-text-line-dataset.pbtxt | 41 …argspec: "args=[\'datasets\', \'choice_dataset\', \'stop_on_empty_dataset\'], varargs=None, keywor…
|
| H A D | tensorflow.data.-t-f-record-dataset.pbtxt | 40 …argspec: "args=[\'datasets\', \'choice_dataset\', \'stop_on_empty_dataset\'], varargs=None, keywor…
|
| H A D | tensorflow.data.experimental.pbtxt | 101 …argspec: "args=[\'datasets\', \'choice_dataset\', \'stop_on_empty_dataset\'], varargs=None, keywor…
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/ |
| H A D | tensorflow.data.-dataset.pbtxt | 52 …argspec: "args=[\'datasets\', \'choice_dataset\', \'stop_on_empty_dataset\'], varargs=None, keywor…
|
| H A D | tensorflow.data.-text-line-dataset.pbtxt | 54 …argspec: "args=[\'datasets\', \'choice_dataset\', \'stop_on_empty_dataset\'], varargs=None, keywor…
|
| H A D | tensorflow.data.experimental.-csv-dataset.pbtxt | 54 …argspec: "args=[\'datasets\', \'choice_dataset\', \'stop_on_empty_dataset\'], varargs=None, keywor…
|
| H A D | tensorflow.data.experimental.pbtxt | 129 …argspec: "args=[\'datasets\', \'choice_dataset\', \'stop_on_empty_dataset\'], varargs=None, keywor…
|
| H A D | tensorflow.data.-fixed-length-record-dataset.pbtxt | 54 …argspec: "args=[\'datasets\', \'choice_dataset\', \'stop_on_empty_dataset\'], varargs=None, keywor…
|
| H A D | tensorflow.data.experimental.-sql-dataset.pbtxt | 54 …argspec: "args=[\'datasets\', \'choice_dataset\', \'stop_on_empty_dataset\'], varargs=None, keywor…
|
| H A D | tensorflow.data.experimental.-random-dataset.pbtxt | 54 …argspec: "args=[\'datasets\', \'choice_dataset\', \'stop_on_empty_dataset\'], varargs=None, keywor…
|
| H A D | tensorflow.data.-t-f-record-dataset.pbtxt | 54 …argspec: "args=[\'datasets\', \'choice_dataset\', \'stop_on_empty_dataset\'], varargs=None, keywor…
|