Searched refs:ReductionToOneDevice (Results 1 – 7 of 7) sorted by relevance
1 path: "tensorflow.distribute.ReductionToOneDevice"3 is_instance: "<class \'tensorflow.python.distribute.cross_device_ops.ReductionToOneDevice\'>"
36 name: "ReductionToOneDevice"
363 class ReductionToOneDevice(CrossDeviceOps): class380 super(ReductionToOneDevice, self).__init__()664 self._simple_cross_replica_ops = ReductionToOneDevice()1108 return ReductionToOneDevice()1113 return ReductionToOneDevice()
107 cross_device_ops_lib.ReductionToOneDevice(reduce_to_device=_LOCAL_CPU))
209 cross_device_ops_lib.ReductionToOneDevice()),212 cross_device_ops_lib.ReductionToOneDevice(216 cross_device_ops_lib.ReductionToOneDevice(314 cross_device_ops_lib.ReductionToOneDevice()),