Searched refs:parameter_devices (Results 1 – 11 of 11) sorted by relevance
10 name: "parameter_devices"
19 name: "parameter_devices"
116 devices = distribution.extended.parameter_devices
441 def parameter_devices(self): member in OneDeviceExtended
666 def parameter_devices(self): member in ParameterServerStrategyExtended
2591 def parameter_devices(self): member in StrategyExtendedV23508 def parameter_devices(self): member in _DefaultDistributionExtended
817 def parameter_devices(self): member in MirroredExtended
1384 def parameter_devices(self): member in TPUExtended
225 num_shards = len(d.extended.parameter_devices)
597 if len(distribution.extended.parameter_devices) != 2:
228 get_expected_variables(len(distribution.extended.parameter_devices)),