| D | values.py | 235 def __init__(self, device_map, values, logical_device=None): argument
272 def device_map(self): member in DistributedValues
444 def __init__(self, strategy, device_map, values, logical_device=None): argument
652 self, strategy, device_map, values, aggregation, logical_device=None): argument
774 self, strategy, device_map, values, aggregation, logical_device=None): argument
825 def device_map(self): member in TPUMirroredVariable
1232 self, strategy, device_map, values, aggregation, logical_device=None): argument
1300 def regroup(device_map, values, wrap_class=PerReplica): argument
1408 def update_regroup(extended, device_map, updates, group): argument
|