Home
last modified time | relevance | path

Searched refs:AutoShardPolicy (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/python/data/experimental/ops/
Ddistribute_options.py28 class AutoShardPolicy(enum.IntEnum): class
43 return dataset_options_pb2.AutoShardPolicy.OFF
45 return dataset_options_pb2.AutoShardPolicy.FILE
47 return dataset_options_pb2.AutoShardPolicy.DATA
49 return dataset_options_pb2.AutoShardPolicy.AUTO
56 if pb == dataset_options_pb2.AutoShardPolicy.OFF:
58 if pb == dataset_options_pb2.AutoShardPolicy.FILE:
60 if pb == dataset_options_pb2.AutoShardPolicy.DATA:
62 if pb == dataset_options_pb2.AutoShardPolicy.AUTO:
121 ty=AutoShardPolicy,
[all …]
Ddata_service_ops.py26 from tensorflow.python.data.experimental.ops.distribute_options import AutoShardPolicy
323 options.experimental_distribute.auto_shard_policy = AutoShardPolicy.OFF
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.data.experimental.-auto-shard-policy.pbtxt1 path: "tensorflow.data.experimental.AutoShardPolicy"
3 is_instance: "<enum \'AutoShardPolicy\'>"
6 mtype: "<enum \'AutoShardPolicy\'>"
10 mtype: "<enum \'AutoShardPolicy\'>"
14 mtype: "<enum \'AutoShardPolicy\'>"
18 mtype: "<enum \'AutoShardPolicy\'>"
Dtensorflow.data.experimental.pbtxt8 name: "AutoShardPolicy"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.data.experimental.-auto-shard-policy.pbtxt1 path: "tensorflow.data.experimental.AutoShardPolicy"
3 is_instance: "<enum \'AutoShardPolicy\'>"
6 mtype: "<enum \'AutoShardPolicy\'>"
10 mtype: "<enum \'AutoShardPolicy\'>"
14 mtype: "<enum \'AutoShardPolicy\'>"
18 mtype: "<enum \'AutoShardPolicy\'>"
Dtensorflow.data.experimental.pbtxt8 name: "AutoShardPolicy"
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dauto_shard.cc124 AutoShardPolicy policy, int64 num_replicas,
628 AutoShardPolicy policy, int64 num_replicas, in OptimizeGraph()
630 if (policy == AutoShardPolicy::OFF || in OptimizeGraph()
631 (policy == AutoShardPolicy::FILE && num_workers == 1 && index == 0)) { in OptimizeGraph()
643 case AutoShardPolicy::OFF: in OptimizeGraph()
646 case AutoShardPolicy::FILE: in OptimizeGraph()
649 case AutoShardPolicy::DATA: in OptimizeGraph()
652 case AutoShardPolicy::AUTO: in OptimizeGraph()
686 AutoShardPolicy(config->parameter_map().at(kAutoShardPolicyAttrName).i()); in Init()
689 if (auto_shard_policy_ != AutoShardPolicy::OFF && in Init()
[all …]
Dauto_shard.h24 enum class AutoShardPolicy { OFF = -1, AUTO = 0, FILE = 1, DATA = 2 }; enum
53 AutoShardPolicy auto_shard_policy_;
/external/tensorflow/tensorflow/python/distribute/
Dinput_ops.py22 from tensorflow.python.data.experimental.ops.distribute_options import AutoShardPolicy
49 AutoShardPolicy.OFF):
Dinput_lib_test.py29 from tensorflow.python.data.experimental.ops.distribute_options import AutoShardPolicy
921 auto_shard_policy=[AutoShardPolicy.AUTO, AutoShardPolicy.OFF]))
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]))
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dauto_shard_dataset_test.py296 distribute_options.AutoShardPolicy.DATA,
297 distribute_options.AutoShardPolicy.AUTO
313 sharding_policy=[distribute_options.AutoShardPolicy.DATA,
314 distribute_options.AutoShardPolicy.FILE]),
348 distribute_options.AutoShardPolicy.DATA)
367 distribute_options.AutoShardPolicy.OFF)
385 distribute_options.AutoShardPolicy.FILE)
633 distribute_options.AutoShardPolicy.DATA,
634 distribute_options.AutoShardPolicy.AUTO
/external/tensorflow/tensorflow/core/framework/
Ddataset_options.proto6 enum AutoShardPolicy { enum
27 AutoShardPolicy auto_shard_policy = 1;
/external/tensorflow/tensorflow/python/data/experimental/
D__init__.py108 from tensorflow.python.data.experimental.ops.distribute_options import AutoShardPolicy
/external/tensorflow/tensorflow/python/data/kernel_tests/
Doptions_test.py139 distribute_options.AutoShardPolicy.DATA)
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_utils_v1.py1685 distribute_options.AutoShardPolicy.OFF)):
Dtraining.py1650 data_option = distribute_options.AutoShardPolicy.DATA
/external/tensorflow/tensorflow/python/keras/distribute/
Ddistribute_strategy_test.py1642 distribute_options.AutoShardPolicy.FILE
/external/tensorflow/
DRELEASE.md1533 `tf.data.experimental.AutoShardPolicy(OFF, AUTO, FILE, DATA)` 2.