Searched refs:AllReduceCrossDeviceOps (Results 1 – 6 of 6) sorted by relevance
4 is_instance: "<class \'tensorflow.python.distribute.cross_device_ops.AllReduceCrossDeviceOps\'>"
304 cross_device_ops_lib.AllReduceCrossDeviceOps("nccl", 1)),307 cross_device_ops_lib.AllReduceCrossDeviceOps("nccl", 0)),323 cross_device_ops_lib.AllReduceCrossDeviceOps393 cross_device_ops_lib.AllReduceCrossDeviceOps())414 cross_device_ops_lib.AllReduceCrossDeviceOps("ring"))
811 class AllReduceCrossDeviceOps(CrossDeviceOps): class834 super(AllReduceCrossDeviceOps, self).__init__()928 AllReduceCrossTowerOps = AllReduceCrossDeviceOps936 class NcclAllReduce(AllReduceCrossDeviceOps):973 class HierarchicalCopyAllReduce(AllReduceCrossDeviceOps):