Searched refs:parameter_devices (Results 1 – 9 of 9) sorted by relevance
18 name: "parameter_devices"
183 def parameter_devices(self): member in OneDeviceExtended
1304 def parameter_devices(self): member in DistributionStrategyExtended1644 def parameter_devices(self): member in _DefaultDistributionExtended
525 def parameter_devices(self): member in ParameterServerStrategyExtended
578 def parameter_devices(self): member in TPUExtended
765 def parameter_devices(self): member in MirroredExtended
191 len(distribution.extended.parameter_devices)),
255 num_shards = len(d.extended.parameter_devices)