| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | auto_shard_dataset_op.cc | 47 int64 index, num_workers, auto_shard_policy, num_replicas; in MakeDataset() local
57 auto_shard_policy = auto_shard_policy_; in MakeDataset()
60 auto config_factory = [num_workers, index, auto_shard_policy, in MakeDataset()
62 return CreateConfig(num_workers, index, auto_shard_policy, num_replicas); in MakeDataset()
74 int64 auto_shard_policy, in CreateConfig() argument
87 {kAutoShardPolicy, auto_shard_policy}, in CreateConfig()
|
| D | auto_shard_dataset_op_test.cc | 28 int auto_shard_policy, int64 num_replicas, in AutoShardDatasetParams() argument
37 auto_shard_policy_(auto_shard_policy) { in AutoShardDatasetParams()
|
| D | auto_shard_dataset_op.h | 43 int64 auto_shard_policy,
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | distribute_options.py | 119 auto_shard_policy = options.create_option( variable in DistributeOptions
148 …pb.auto_shard_policy = AutoShardPolicy._to_proto(self.auto_shard_policy) # pylint: disable=protec…
154 …self.auto_shard_policy = AutoShardPolicy._from_proto(pb.auto_shard_policy) # pylint: disable=prot…
|
| D | distribute.py | 77 auto_shard_policy=int(
78 input_dataset.options().experimental_distribute.auto_shard_policy),
|
| D | data_service_ops.py | 323 options.experimental_distribute.auto_shard_policy = AutoShardPolicy.OFF
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | AutoShardDataset.pbtxt | 20 name: "auto_shard_policy"
58 name: "auto_shard_policy"
|
| D | ExperimentalAutoShardDataset.pbtxt | 51 name: "auto_shard_policy"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.data.experimental.-distribute-options.pbtxt | 7 name: "auto_shard_policy"
|
| D | tensorflow.raw_ops.pbtxt | 269 … \'num_workers\', \'index\', \'output_types\', \'output_shapes\', \'auto_shard_policy\', \'num_rep…
1405 … \'num_workers\', \'index\', \'output_types\', \'output_shapes\', \'auto_shard_policy\', \'name\']…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.data.experimental.-distribute-options.pbtxt | 7 name: "auto_shard_policy"
|
| D | tensorflow.raw_ops.pbtxt | 269 … \'num_workers\', \'index\', \'output_types\', \'output_shapes\', \'auto_shard_policy\', \'num_rep…
1405 … \'num_workers\', \'index\', \'output_types\', \'output_shapes\', \'auto_shard_policy\', \'name\']…
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | input_lib_test.py | 921 auto_shard_policy=[AutoShardPolicy.AUTO, AutoShardPolicy.OFF]))
923 iteration_type, auto_shard_policy): argument
929 ds_option.experimental_distribute.auto_shard_policy = auto_shard_policy
936 if auto_shard_policy == AutoShardPolicy.AUTO:
1301 options.experimental_distribute.auto_shard_policy = AutoShardPolicy.OFF
1369 options.experimental_distribute.auto_shard_policy = AutoShardPolicy.OFF
1404 auto_shard_policy=[AutoShardPolicy.AUTO, AutoShardPolicy.DATA]))
1406 distribution, auto_shard_policy): argument
1422 options.experimental_distribute.auto_shard_policy = auto_shard_policy
|
| D | input_ops.py | 48 if (dataset.options().experimental_distribute.auto_shard_policy !=
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | ExperimentalAutoShardDataset.pbtxt | 51 name: "auto_shard_policy"
|
| D | AutoShardDataset.pbtxt | 51 name: "auto_shard_policy"
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | auto_shard_dataset_test.py | 304 options.experimental_distribute.auto_shard_policy = sharding_policy
326 options.experimental_distribute.auto_shard_policy = sharding_policy
347 options.experimental_distribute.auto_shard_policy = (
366 options.experimental_distribute.auto_shard_policy = (
384 options.experimental_distribute.auto_shard_policy = (
643 options.experimental_distribute.auto_shard_policy = sharding_policy
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | options_test.py | 138 options.experimental_distribute.auto_shard_policy = (
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | dataset_options.proto | 27 AutoShardPolicy auto_shard_policy = 1; field
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training_utils_v1.py | 1684 (dataset.options().experimental_distribute.auto_shard_policy !=
|
| D | training.py | 1651 options.experimental_distribute.auto_shard_policy = data_option
|
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | distribute_strategy_test.py | 1641 options.experimental_distribute.auto_shard_policy = \
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 2824 name: "auto_shard_policy"
14625 name: "auto_shard_policy"
|