Searched defs:sharding_policy (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/service/ |
| D | auto_shard_test.py | 68 def testRangeDataset_AutoShard(self, sharding_policy): argument
146 self, sharding_policy): argument
204 def testTFRecordDataset_AutoShard(self, sharding_policy): argument
227 def testTFRecordDataset_ShuffleFileList(self, sharding_policy): argument
340 def testTFRecordDataset_FewerFilesThanWorkers(self, sharding_policy): argument
396 def testBatchDataset(self, sharding_policy): argument
569 def testEnumerateShardingPolicies(self, sharding_policy): argument
|
| /external/tensorflow/tensorflow/core/data/service/ |
| D | test_cluster.h | 142 ProcessingModeDef::ShardingPolicy sharding_policy, in Read()
162 const int64 dataset_id, ProcessingModeDef::ShardingPolicy sharding_policy, in CreateJob()
|
| 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 | 64 const ProcessingModeDef::ShardingPolicy sharding_policy) { in ToAutoShardPolicy()
|
| D | auto_shard_rewriter_test.cc | 75 TaskDef GetTaskDef(const ProcessingModeDef::ShardingPolicy sharding_policy, in GetTaskDef()
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | data_service.proto | 47 ShardingPolicy sharding_policy = 1; field
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | auto_shard_dataset_test.py | 300 def testShardByDataBeforePrefetch(self, sharding_policy): argument
317 def testReplicateAndShardProduceDisjointData(self, shuffle, sharding_policy): argument
623 def testUseLegacyRebatchWithDataSharding(self, sharding_policy, argument
|