Searched refs:auto_shard_dataset (Results 1 – 4 of 4) sorted by relevance
92 auto_shard_dataset=False, argument102 auto_shard_dataset)169 auto_shard_dataset=False): argument177 self._auto_shard_dataset = auto_shard_dataset
105 dataset = input_ops.auto_shard_dataset(115 dataset = input_ops.auto_shard_dataset(126 dataset = input_ops.auto_shard_dataset(139 dataset = input_ops.auto_shard_dataset(171 dataset = input_ops.auto_shard_dataset(199 dataset = input_ops.auto_shard_dataset(210 dataset = input_ops.auto_shard_dataset(229 dataset = input_ops.auto_shard_dataset(238 dataset = input_ops.auto_shard_dataset(247 dataset = input_ops.auto_shard_dataset([all …]
36 def auto_shard_dataset(dataset, num_shards, index): function
231 `input_fn` if `auto_shard_dataset` is set to `True`. As a result, each worker