Searched defs:sharding_policy (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/service/ |
| D | auto_shard_test.py | 73 def testRangeDataset_AutoShard(self, sharding_policy): argument
145 self, sharding_policy): argument
198 def testTFRecordDataset_AutoShard(self, sharding_policy): argument
222 def testTFRecordDataset_ShuffleFileList(self, sharding_policy): argument
333 def testTFRecordDataset_FewerFilesThanWorkers(self, sharding_policy): argument
385 def testBatchDataset(self, sharding_policy): argument
564 def testEnumerateShardingPolicies(self, sharding_policy): argument
|
| D | metadata_test.py | 104 def testCardinality(self, dataset_fn, sharding_policy, expected_result): argument
112 def testFromDatasetIdCardinality(self, dataset_fn, sharding_policy, argument
|
| D | coordinated_read_ft_test.py | 91 def testMultiStartStop(self, sharding_policy): argument
|
| D | test_base.py | 292 sharding_policy=data_service_ops.ShardingPolicy.OFF): argument
|
| /external/tensorflow/tensorflow/core/data/service/ |
| D | test_cluster.h | 169 ProcessingModeDef::ShardingPolicy sharding_policy, in Read()
193 ProcessingModeDef::ShardingPolicy sharding_policy, in CreateIteration()
|
| D | common_test.cc | 98 for (const ProcessingModeDef::ShardingPolicy sharding_policy : in TEST() local
105 const ProcessingModeDef::ShardingPolicy sharding_policy = in TEST() local
|
| D | common.cc | 68 const ProcessingModeDef::ShardingPolicy sharding_policy) { in ToAutoShardPolicy()
|
| D | auto_shard_rewriter_test.cc | 72 TaskDef GetTaskDef(const ProcessingModeDef::ShardingPolicy sharding_policy, in GetTaskDef()
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | auto_shard_dataset_test.py | 326 def testShardByDataBeforePrefetch(self, sharding_policy): argument
343 def testReplicateAndShardProduceDisjointData(self, shuffle, sharding_policy): argument
649 def testUseLegacyRebatchWithDataSharding(self, sharding_policy, argument
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | data_service.proto | 48 ShardingPolicy sharding_policy = 1; field
|