Searched refs:auto_shard_dataset (Results 1 – 5 of 5) sorted by relevance
111 dataset = input_ops.auto_shard_dataset(121 dataset = input_ops.auto_shard_dataset(132 dataset = input_ops.auto_shard_dataset(145 dataset = input_ops.auto_shard_dataset(175 dataset = input_ops.auto_shard_dataset(202 dataset = input_ops.auto_shard_dataset(214 dataset = input_ops.auto_shard_dataset(233 dataset = input_ops.auto_shard_dataset(241 dataset = input_ops.auto_shard_dataset(250 dataset = input_ops.auto_shard_dataset(
26 def auto_shard_dataset(dataset, num_shards, index, num_replicas_in_sync=None): function
1144 dataset = input_ops.auto_shard_dataset(dataset,1157 cloned_dataset = input_ops.auto_shard_dataset(
233 Node* auto_shard_dataset; in TEST_F() local273 .Finalize(&graph, &auto_shard_dataset)); in TEST_F()275 add_identity_nodes(auto_shard_dataset, graph, identity_nodes); in TEST_F()
74 variant_tensor = ged_ops.auto_shard_dataset(