Home
last modified time | relevance | path

Searched defs:auto_shard_policy (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dauto_shard_dataset_op.cc47 int64 index, num_workers, auto_shard_policy, num_replicas; in MakeDataset() local
74 int64 auto_shard_policy, in CreateConfig()
Dauto_shard_dataset_op_test.cc28 int auto_shard_policy, int64 num_replicas, in AutoShardDatasetParams()
/external/tensorflow/tensorflow/python/data/experimental/ops/
Ddistribute_options.py119 auto_shard_policy = options.create_option( variable in DistributeOptions
/external/tensorflow/tensorflow/core/framework/
Ddataset_options.proto27 AutoShardPolicy auto_shard_policy = 1; field
/external/tensorflow/tensorflow/python/distribute/
Dinput_lib_test.py923 iteration_type, auto_shard_policy): argument
1406 distribution, auto_shard_policy): argument