Home
last modified time | relevance | path

Searched refs:collective_util (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dcollective_util_test.py21 from tensorflow.python.distribute import collective_util
28 options = collective_util._OptionsExported(bytes_per_pack=1)
29 self.assertIsInstance(options, collective_util.Options)
32 collective_util._OptionsExported(bytes_per_pack=-1)
36 options = collective_util.Hints(50, 1)
38 self.assertIsInstance(options, collective_util.Options)
Dcollective_all_reduce_strategy.py28 from tensorflow.python.distribute import collective_util
184 communication_options = collective_util.Options()
239 communication=collective_util.CommunicationImplementation.AUTO,
252 communication_options = collective_util.Options(
261 communication=collective_util.CommunicationImplementation.AUTO):
280 communication=collective_util.CommunicationImplementation.AUTO,
283 communication_options = collective_util.Options(
318 if not isinstance(communication_options, collective_util.Options):
704 collective_util.CommunicationImplementation.NCCL):
780 options = collective_util.Options()
[all …]
Dcross_device_ops_test.py31 from tensorflow.python.distribute import collective_util
56 CommunicationImplementation = collective_util.CommunicationImplementation
222 collective_util.Options(), True)
335 communication_options=collective_util.Options(
378 communication_options=collective_util.Options(
459 communication_options=collective_util.Options(
504 communication_options=collective_util.Options(
587 options = collective_util.Options(implementation=implementation)
646 options = collective_util.Options(implementation=implementation)
686 options = collective_util.Options(implementation=implementation)
[all …]
Dcross_device_ops.py30 from tensorflow.python.distribute import collective_util
284 options = collective_util.Options()
299 options = collective_util.Options()
332 options = collective_util.Options()
404 options = collective_util.Options()
426 options = collective_util.Options()
1016 CollectiveCommunication = collective_util.CommunicationImplementation
1017 CommunicationImplementation = collective_util.CommunicationImplementation
Ddistribute_lib.py203 from tensorflow.python.distribute import collective_util
2299 options = collective_util.Options()
2377 options = collective_util.Options()
2413 options = collective_util.Options()
2455 options = collective_util.Options()
2464 options = collective_util.Options()
3119 options = collective_util.Options()
3286 options = collective_util.Options()
Dmirrored_strategy.py23 from tensorflow.python.distribute import collective_util
331 self._communication_options = collective_util.Options(
332 implementation=collective_util.CommunicationImplementation.NCCL)
334 self._communication_options = collective_util.Options()
DBUILD35 ":collective_util",
137 ":collective_util",
315 ":collective_util",
408 ":collective_util",
691 name = "collective_util",
692 srcs = ["collective_util.py"],
704 ":collective_util",
1097 ":collective_util",
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.distribute.experimental.-collective-hints.pbtxt3 is_instance: "<class \'tensorflow.python.distribute.collective_util.Hints\'>"
Dtensorflow.distribute.experimental.-communication-options.pbtxt3 is_instance: "<class \'tensorflow.python.distribute.collective_util._OptionsExported\'>"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.distribute.experimental.-collective-hints.pbtxt3 is_instance: "<class \'tensorflow.python.distribute.collective_util.Hints\'>"
Dtensorflow.distribute.experimental.-communication-options.pbtxt3 is_instance: "<class \'tensorflow.python.distribute.collective_util._OptionsExported\'>"
/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_util.h28 namespace collective_util {
Dring_reducer.cc257 Status s = collective_util::ComputeBinOp( in RunAsyncParts()
272 Status s = collective_util::ComputeBinOp( in RunAsyncParts()
Dcollective_util.cc29 namespace collective_util { namespace
Dpermuter.cc58 return collective_util::InitializeDeviceAndLocality( in InitializeCollectiveContext()
Dhierarchical_tree_broadcaster.cc183 VLOG(2) << collective_util::SubdivPermDebugString(*col_params); in InitializeCollectiveParams()
192 return collective_util::InitializeDeviceAndLocality( in InitializeCollectiveContext()
DBUILD221 "collective_util.h",
349 name = "collective_util",
350 srcs = ["collective_util.cc"],
351 hdrs = ["collective_util.h"],
830 ":collective_util",
1133 ":collective_util",
1323 ":collective_util",
1342 ":collective_util",
1364 ":collective_util",
1577 ":collective_util",
Dring_alg.cc240 VLOG(2) << collective_util::SubdivPermDebugString(*col_params); in InitializeCollectiveParams()
249 return collective_util::InitializeDeviceAndLocality( in InitializeCollectiveContext()
/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl.cc65 return collective_util::InitializeDeviceAndLocality( in InitializeCollectiveContext()
Dcollective_nccl_reducer.cc117 final_status = collective_util::ComputeBinOp( in Run()
/external/tensorflow/tensorflow/python/distribute/v1/
Dcross_device_ops_test.py31 from tensorflow.python.distribute import collective_util
435 CollectiveCommunication = collective_util.CollectiveCommunication
473 comm_options = collective_util.Options(implementation=communication)
504 comm_options = collective_util.Options(implementation=communication)
731 hints = collective_util.Hints(bytes_per_pack=bytes_per_pack)
DBUILD25 "//tensorflow/python/distribute:collective_util",