Searched refs:NcclAllReduce (Results 1 – 17 of 17) sorted by relevance
1 path: "tensorflow.distribute.NcclAllReduce"3 is_instance: "<class \'tensorflow.python.distribute.cross_device_ops.NcclAllReduce\'>"
44 name: "NcclAllReduce"
2612 name: "NcclAllReduce"
24 name: "NcclAllReduce"
2 graph_op_name: "NcclAllReduce"
2 name: "NcclAllReduce"
74 "nccl": tf.distribute.NcclAllReduce,
356 isinstance(self._cross_device_ops, cross_device_ops_lib.NcclAllReduce)358 cross_device_ops_lib.NcclAllReduce)):437 self._inferred_cross_device_ops = cross_device_ops_lib.NcclAllReduce()
936 class NcclAllReduce(AllReduceCrossDeviceOps): class968 super(NcclAllReduce, self).__init__(1420 return NcclAllReduce(num_packs=1)
309 cross_device_ops_lib.NcclAllReduce()),356 cross_device_ops_lib.NcclAllReduce)
28753 func NcclAllReduce(scope *Scope, input tf.Output, reduction string, num_devices int64, shared_name … func
26080 name: "NcclAllReduce"
40433 name: "NcclAllReduce"