_AutoShardDataset (reference) in projects: external

Searched refs:_AutoShardDataset (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
D	auto_shard_dataset_test.py	97 dataset = distribute._AutoShardDataset(dataset, 5, 3) 134 dataset = distribute._AutoShardDataset(dataset, 5, 0) 155 dataset = distribute._AutoShardDataset(dataset, 5, 3) 182 dataset = distribute._AutoShardDataset(dataset, 5, 3) 202 dataset = distribute._AutoShardDataset(dataset, 5, 3) 214 dataset = distribute._AutoShardDataset(dataset, 5, 0) 236 dataset = distribute._AutoShardDataset(dataset, 5, 3) 260 dataset = distribute._AutoShardDataset(dataset, 2, index) 283 dataset = distribute._AutoShardDataset(dataset, 5, 3) 306 dataset = distribute._AutoShardDataset(dataset, 2, 0) [all …]
/external/tensorflow/tensorflow/python/data/experimental/ops/
D	distribute.py	`34 class _AutoShardDataset(dataset_ops.UnaryDataset): class 81 super(_AutoShardDataset, self).__init__(input_dataset, variant_tensor) 90 _AutoShardDataset(input_dataset, num_workers, index, num_replicas)) 527 _AutoShardDatasetV1.__doc__ = _AutoShardDataset.__doc__`
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/
D	auto_shard_dataset_serialization_test.py	`64 dataset = distribute._AutoShardDataset(dataset, 5, 3)`
/external/tensorflow/tensorflow/python/distribute/
D	input_ops.py	`56 return distribute._AutoShardDataset(dataset, num_shards, index,`