Home
last modified time | relevance | path

Searched refs:enable_collective_ops (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/
Dcollective_ops_multi_worker_test.py38 def enable_collective_ops(cluster_resolver): function
48 context.context().enable_collective_ops(server_def)
53 enable_collective_ops(cluster_resolver)
68 enable_collective_ops(cluster_resolver_lib.TFConfigClusterResolver())
92 enable_collective_ops(cluster_resolver_lib.TFConfigClusterResolver())
107 enable_collective_ops(cluster_resolver)
144 enable_collective_ops(cluster_resolver_lib.TFConfigClusterResolver())
157 initializer=lambda: enable_collective_ops(cluster_resolver_lib.
/external/tensorflow/tensorflow/dtensor/python/
Dmulti_client_util.py107 context.context().enable_collective_ops(server_def)
/external/tensorflow/tensorflow/python/distribute/
Ddevice_util_test.py89 context.context().enable_collective_ops(server_def)
Dcross_device_ops_test.py91 def enable_collective_ops(): function
105 context.context().enable_collective_ops(server_def)
149 global_mpr_1p.runner.run(enable_collective_ops)
150 global_mpr_2p.runner.run(enable_collective_ops)
Dcollective_all_reduce_strategy.py519 context.context().enable_collective_ops(server_def)
/external/tensorflow/tensorflow/python/eager/
Dcontext.py800 def enable_collective_ops(self, server_def): member in Context