Searched refs:auto_shard_dataset (Results 1 – 4 of 4) sorted by relevance
115 dataset = input_ops.auto_shard_dataset(125 dataset = input_ops.auto_shard_dataset(136 dataset = input_ops.auto_shard_dataset(149 dataset = input_ops.auto_shard_dataset(179 dataset = input_ops.auto_shard_dataset(206 dataset = input_ops.auto_shard_dataset(218 dataset = input_ops.auto_shard_dataset(237 dataset = input_ops.auto_shard_dataset(245 dataset = input_ops.auto_shard_dataset(254 dataset = input_ops.auto_shard_dataset(
30 def auto_shard_dataset(dataset, num_shards, index, num_replicas_in_sync=None): function
1014 dataset = input_ops.auto_shard_dataset(dataset,1028 cloned_dataset = input_ops.auto_shard_dataset(
73 variant_tensor = ged_ops.auto_shard_dataset(