Searched defs:auto_shard_policy (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | auto_shard_dataset_op.cc | 47 int64_t index, num_workers, auto_shard_policy, num_replicas; in MakeDataset() local
80 int64_t auto_shard_policy, in CreateConfig()
|
| D | auto_shard_dataset_op_test.cc | 28 int64_t index, int auto_shard_policy, in AutoShardDatasetParams()
|
| /external/tensorflow/tensorflow/core/data/service/ |
| D | auto_shard_rewriter.cc | 136 AutoShardRewriter::AutoShardRewriter(AutoShardPolicy auto_shard_policy, in AutoShardRewriter()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | dataset_options.proto | 50 AutoShardPolicy auto_shard_policy = 1; field
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | options.py | 202 auto_shard_policy = options_lib.create_option( variable in DistributeOptions
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | auto_shard_dataset_test.py | 566 def testEnumerateAutoShardPolicies(self, auto_shard_policy): argument
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | input_lib_test.py | 945 iteration_type, auto_shard_policy): argument
1429 distribution, auto_shard_policy): argument
|