Home
last modified time | relevance | path

Searched refs:communication_options (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dcollective_all_reduce_strategy.py169 communication_options=None): argument
183 if communication_options is None:
184 communication_options = collective_util.Options()
189 communication_options=communication_options))
200 def _from_local_devices(cls, devices, communication_options=None): argument
202 obj = cls(communication_options=communication_options)
252 communication_options = collective_util.Options(
255 self).__init__(cluster_resolver, communication_options)
283 communication_options = collective_util.Options(
289 communication_options=communication_options))
[all …]
Dcross_device_ops_test.py246 options.communication_options)
292 options.communication_options)
335 communication_options=collective_util.Options(
378 communication_options=collective_util.Options(
459 communication_options=collective_util.Options(
504 communication_options=collective_util.Options(
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.distribute.-multi-worker-mirrored-strategy.pbtxt21 …argspec: "args=[\'self\', \'cluster_resolver\', \'communication_options\'], varargs=None, keywords…
/external/tensorflow/tensorflow/python/distribute/v1/
Dcross_device_ops_test.py506 communication_options=comm_options, cluster_resolver=resolver)