Searched refs:replica_device_setter (Results 1 – 6 of 6) sorted by relevance
40 device_setter.replica_device_setter(cluster=self._cluster_spec)):55 device_setter.replica_device_setter(cluster=self._cluster_spec)):62 device_setter.replica_device_setter(cluster=self._cluster_spec)):75 device_setter.replica_device_setter(cluster=self._cluster_spec)):93 device_setter.replica_device_setter(ps_tasks=1, ps_device="/cpu:0")):107 device_setter.replica_device_setter(122 device_setter.replica_device_setter(cluster=self._cluster_spec.as_dict(136 device_setter.replica_device_setter(155 device_setter.replica_device_setter(176 device_setter.replica_device_setter(
137 def replica_device_setter(ps_tasks=0, function
95 with ops.device(device_setter.replica_device_setter(ps_tasks=100)):
81 from tensorflow.python.training.device_setter import replica_device_setter
251 self._variable_device = device_setter.replica_device_setter(
428 name: "replica_device_setter"