Searched refs:_get_replica_device_setter (Results 1 – 3 of 3) sorted by relevance
1337 with ops.device(estimator._get_replica_device_setter(config)):1349 estimator._get_replica_device_setter(run_config.RunConfig())):1366 with ops.device(estimator._get_replica_device_setter(config)):1376 estimator._get_replica_device_setter(run_config.RunConfig())):1399 with ops.device(estimator._get_replica_device_setter(config)):
209 def _get_replica_device_setter(config): function459 self._device_fn = _get_replica_device_setter(self._config)
728 with ops.device(self._get_replica_device_setter(worker_device)):1279 def _get_replica_device_setter(self, worker_device): member in GradientBoostedDecisionTreeModel