Home
last modified time | relevance | path

Searched refs:replica_device_setter (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/python/training/
Ddevice_setter_test.py40 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(
Ddevice_setter.py134 def replica_device_setter(ps_tasks=0, ps_device="/job:ps", function
Dlocalhost_cluster_performance_test.py95 with ops.device(device_setter.replica_device_setter(ps_tasks=100)):
Dtraining.py78 from tensorflow.python.training.device_setter import replica_device_setter
/external/tensorflow/tensorflow/contrib/training/python/training/
Ddevice_setter_test.py42 device_setter.replica_device_setter(
76 device_setter.replica_device_setter(
97 device_setter.replica_device_setter(
118 device_setter.replica_device_setter(
/external/tensorflow/tensorflow/contrib/opt/python/training/
Delastic_average_optimizer_test.py80 device_setter.replica_device_setter(
93 device_setter.replica_device_setter(
283 device_setter.replica_device_setter(cluster=cluster_spec,
Dmodel_average_optimizer_test.py74 device_setter.replica_device_setter(
185 device_setter.replica_device_setter(cluster=cluster_spec,
Dagn_optimizer_test.py73 ps_device = device_setter.replica_device_setter(
267 device_setter.replica_device_setter(cluster=cluster_spec,
/external/tensorflow/tensorflow/python/distribute/
Dparameter_server_strategy.py176 self._variable_device = device_setter.replica_device_setter(
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.train.pbtxt420 name: "replica_device_setter"
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Destimator.py231 return device_setter.replica_device_setter(
/external/tensorflow/tensorflow/contrib/boosted_trees/python/training/functions/
Dgbdt_batch.py1289 return device_setter.replica_device_setter(
/external/tensorflow/tensorflow/contrib/framework/python/ops/
Dvariables_test.py457 with ops.device(device_setter.replica_device_setter(ps_tasks=2)):